Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/clementfarabet/lua---nnx.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--init.lua1
-rw-r--r--nnx-1.0-1.rockspec1
2 files changed, 2 insertions, 0 deletions
diff --git a/init.lua b/init.lua
index 4e694db..9815544 100644
--- a/init.lua
+++ b/init.lua
@@ -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)