From bf7a27b2a6f1de76b72fe5c3b3b75bd92bf856ab Mon Sep 17 00:00:00 2001 From: Clement Farabet Date: Mon, 2 Apr 2012 15:24:29 -0400 Subject: KL Divergence criterion, for continuous distributions. This is the analogous of ClassNLL, but for non 1-of-N distributions. --- init.lua | 1 + 1 file changed, 1 insertion(+) (limited to 'init.lua') diff --git a/init.lua b/init.lua index 4fe448e..d589eec 100644 --- a/init.lua +++ b/init.lua @@ -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') -- cgit v1.2.3