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:
authorClement Farabet <clement.farabet@gmail.com>2011-08-24 00:41:13 +0400
committerClement Farabet <clement.farabet@gmail.com>2011-08-24 00:41:13 +0400
commit1b54395416e9fda3153f8501a7825e847ef96df1 (patch)
tree0cc9bc4df232a32d98effa736d73ca807456b942 /init.lua
parentb03dd22e0afc878d2454f8ed76d94b4ac9957a34 (diff)
Added new Optimization framework to support new algos.
Diffstat (limited to 'init.lua')
-rw-r--r--init.lua4
1 files changed, 4 insertions, 0 deletions
diff --git a/init.lua b/init.lua
index 99e2fa7..3fc2ac1 100644
--- a/init.lua
+++ b/init.lua
@@ -97,6 +97,10 @@ torch.include('nnx', 'SpatialMSECriterion.lua')
torch.include('nnx', 'SpatialClassNLLCriterion.lua')
torch.include('nnx', 'SpatialSparseCriterion.lua')
+-- optimizations:
+torch.include('nnx', 'Optimization.lua')
+torch.include('nnx', 'SGDOptimization.lua')
+
-- trainers:
torch.include('nnx', 'Trainer.lua')
torch.include('nnx', 'StochasticTrainer.lua')