diff options
author | Francisco Massa <fvsmassa@gmail.com> | 2016-01-09 17:31:18 +0300 |
---|---|---|
committer | soumith <soumith@gmail.com> | 2016-10-07 23:08:39 +0300 |
commit | eb7656e621347e3d9150ed593279bae63dd004f4 (patch) | |
tree | 006cabc4d5898714ff80fc5ad9f5bdc91f6774ba /init.lua | |
parent | c7c4fa2c0a39da5087b479233444ba39c1dd80a2 (diff) |
Update LogSoftMax to work in spatial domain
Diffstat (limited to 'init.lua')
-rw-r--r-- | init.lua | 2 |
1 files changed, 2 insertions, 0 deletions
@@ -90,6 +90,7 @@ require('nn.ReLU6') require('nn.PReLU') require('nn.LeakyReLU') require('nn.SpatialSoftMax') +require('nn.SpatialLogSoftMax') require('nn.RReLU') require('nn.ELU') @@ -183,4 +184,5 @@ require('nn.SparseJacobian') require('nn.hessian') require('nn.test') + return nn |