1 files changed, 1 insertions, 0 deletions
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')