diff options
author | Clement Farabet <clement.farabet@gmail.com> | 2012-04-02 23:24:29 +0400 |
---|---|---|
committer | Clement Farabet <clement.farabet@gmail.com> | 2012-04-02 23:24:29 +0400 |
commit | bf7a27b2a6f1de76b72fe5c3b3b75bd92bf856ab (patch) | |
tree | 95a9151b57caf8ebed7c8492696d94c4d36c1ba6 /init.lua | |
parent | de5dc53c9339ad77aee8581904e69ee879a8fe3e (diff) |
KL Divergence criterion, for continuous distributions.
This is the analogous of ClassNLL, but for non 1-of-N
distributions.
Diffstat (limited to 'init.lua')
-rw-r--r-- | init.lua | 1 |
1 files changed, 1 insertions, 0 deletions
@@ -79,6 +79,7 @@ torch.include('nn', 'MSECriterion.lua') torch.include('nn', 'MarginCriterion.lua') torch.include('nn', 'AbsCriterion.lua') torch.include('nn', 'ClassNLLCriterion.lua') +torch.include('nn', 'DistKLDivCriterion.lua') torch.include('nn', 'MultiCriterion.lua') torch.include('nn', 'L1HingeEmbeddingCriterion.lua') torch.include('nn', 'HingeEmbeddingCriterion.lua') |