diff options
| author | Clement Farabet <clement.farabet@gmail.com> | 2011-08-30 00:55:12 +0400 |
|---|---|---|
| committer | Clement Farabet <clement.farabet@gmail.com> | 2011-08-30 00:55:12 +0400 |
| commit | 398e21cfaf6a8e4fa47edd9e6de081497e75dda2 (patch) | |
| tree | 6d9779b6d7fa47cf518993c2d788429369a58264 /init.lua | |
| parent | 0b7f3b9cd4e578a96cfc6b50b18bca2f27cc4682 (diff) | |
Added DistNLLCriterion, to support Neg Likelihood for distributions.
ClassNLLCriterion only supports simple distributions, e.g. one-of-N.
DistNLLCriterion supports arbitrary distributions.
Diffstat (limited to 'init.lua')
| -rw-r--r-- | init.lua | 1 |
1 files changed, 1 insertions, 0 deletions
@@ -93,6 +93,7 @@ torch.include('nnx', 'SpatialColorTransform.lua') -- criterions: torch.include('nnx', 'SuperCriterion.lua') torch.include('nnx', 'SparseCriterion.lua') +torch.include('nnx', 'DistNLLCriterion.lua') torch.include('nnx', 'SpatialMSECriterion.lua') torch.include('nnx', 'SpatialClassNLLCriterion.lua') torch.include('nnx', 'SpatialSparseCriterion.lua') |