diff options
Diffstat (limited to 'nnx-1.0-1.rockspec')
-rw-r--r-- | nnx-1.0-1.rockspec | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/nnx-1.0-1.rockspec b/nnx-1.0-1.rockspec index 4529d24..dcb8d1b 100644 --- a/nnx-1.0-1.rockspec +++ b/nnx-1.0-1.rockspec @@ -62,6 +62,7 @@ build = { install_files(/lua/nnx init.lua) install_files(/lua/nnx Abs.lua) install_files(/lua/nnx ConfusionMatrix.lua) + install_files(/lua/nnx DistNLLCriterion.lua) install_files(/lua/nnx Logger.lua) install_files(/lua/nnx Probe.lua) install_files(/lua/nnx HardShrink.lua) @@ -83,7 +84,6 @@ build = { install_files(/lua/nnx SpatialCriterion.lua) install_files(/lua/nnx Trainer.lua) install_files(/lua/nnx OnlineTrainer.lua) - install_files(/lua/nnx StochasticTrainer.lua) install_files(/lua/nnx DataSet.lua) install_files(/lua/nnx DataList.lua) install_files(/lua/nnx DataSetLabelMe.lua) @@ -103,6 +103,8 @@ build = { install_files(/lua/nnx Optimization.lua) install_files(/lua/nnx LBFGSOptimization.lua) install_files(/lua/nnx SGDOptimization.lua) + install_files(/lua/nnx BatchOptimization.lua) + install_files(/lua/nnx BatchTrainer.lua) add_subdirectory (test) install_targets(/lib nnx) ]], |