diff options
-rw-r--r-- | init.lua | 1 | ||||
-rw-r--r-- | nnx-1.0-1.rockspec | 1 |
2 files changed, 2 insertions, 0 deletions
@@ -93,6 +93,7 @@ torch.include('nnx', 'SpatialConvolutionSparse.lua') torch.include('nnx', 'SuperCriterion.lua') torch.include('nnx', 'SparseCriterion.lua') torch.include('nnx', 'DistNLLCriterion.lua') +torch.include('nnx', 'KLDivCriterion.lua') torch.include('nnx', 'SpatialMSECriterion.lua') torch.include('nnx', 'SpatialClassNLLCriterion.lua') torch.include('nnx', 'SpatialSparseCriterion.lua') diff --git a/nnx-1.0-1.rockspec b/nnx-1.0-1.rockspec index 8489dbf..ccb3e01 100644 --- a/nnx-1.0-1.rockspec +++ b/nnx-1.0-1.rockspec @@ -95,6 +95,7 @@ build = { install_files(/lua/nnx Abs.lua) install_files(/lua/nnx ConfusionMatrix.lua) install_files(/lua/nnx DistNLLCriterion.lua) + install_files(/lua/nnx KLDivCriterion.lua) install_files(/lua/nnx Logger.lua) install_files(/lua/nnx Probe.lua) install_files(/lua/nnx HardShrink.lua) |