1 files changed, 1 insertions, 0 deletions
diff --git a/init.lua b/init.lua
index 20246bc..ea44de0 100644
--- a/init.lua
+++ b/init.lua
@@ -93,6 +93,7 @@ torch.include('nnx', 'SpatialColorTransform.lua')
 -- criterions:
 torch.include('nnx', 'SuperCriterion.lua')
 torch.include('nnx', 'SparseCriterion.lua')
+torch.include('nnx', 'DistNLLCriterion.lua')
 torch.include('nnx', 'SpatialMSECriterion.lua')
 torch.include('nnx', 'SpatialClassNLLCriterion.lua')
 torch.include('nnx', 'SpatialSparseCriterion.lua')