diff options
author | Clement Farabet <clement.farabet@gmail.com> | 2011-07-08 21:55:16 +0400 |
---|---|---|
committer | Clement Farabet <clement.farabet@gmail.com> | 2011-07-08 21:55:16 +0400 |
commit | 836bdb217fb406711283c74e50b220680ba14191 (patch) | |
tree | a832fb983e3e987d2cd718307a99a16d1eebe16c /init.lua | |
parent | f2cd86ebd312900c8de87616114169eedab5721a (diff) |
Added two new spatial criterions.
Diffstat (limited to 'init.lua')
-rw-r--r-- | init.lua | 4 |
1 files changed, 3 insertions, 1 deletions
@@ -82,6 +82,8 @@ torch.include('nnx', 'SpatialFovea.lua') -- criterions: torch.include('nnx', 'SuperCriterion.lua') +torch.include('nnx', 'SpatialMSECriterion.lua') +torch.include('nnx', 'SpatialClassNLLCriterion.lua') -- trainers: torch.include('nnx', 'Trainer.lua') @@ -89,4 +91,4 @@ torch.include('nnx', 'StochasticTrainer.lua') -- datasets: torch.include('nnx', 'DataSet.lua') -torch.include('nnx', 'DataList.lua')
\ No newline at end of file +torch.include('nnx', 'DataList.lua') |