diff options
| author | Sasank Chilamkurthy <sasankchilamkurthy@gmail.com> | 2016-08-25 08:04:53 +0300 |
|---|---|---|
| committer | Sasank Chilamkurthy <sasankchilamkurthy@gmail.com> | 2016-08-25 08:04:53 +0300 |
| commit | 7eeeeed054d31f8a58ddcb8f53a93dcd9dd3ee5b (patch) | |
| tree | 7bca40ccd95d71fc8b3addc428c26d0c22815d41 | |
| parent | 78e291388fd2ba7b4439fb58a0ad3ecb816da0c2 (diff) | |
Add new modules to init.lua
| -rw-r--r-- | init.lua | 3 |
1 files changed, 3 insertions, 0 deletions
@@ -177,6 +177,8 @@ require('cudnn.Tanh') require('cudnn.Sigmoid') require('cudnn.SpatialSoftMax') require('cudnn.SpatialLogSoftMax') +require('cudnn.VolumetricSoftMax') +require('cudnn.VolumetricLogSoftMax') require('cudnn.SoftMax') require('cudnn.LogSoftMax') require('cudnn.SpatialCrossMapLRN') @@ -184,6 +186,7 @@ require('cudnn.BatchNormalization') require('cudnn.SpatialBatchNormalization') require('cudnn.VolumetricBatchNormalization') require('cudnn.SpatialCrossEntropyCriterion') +require('cudnn.VolumetricCrossEntropyCriterion') require('cudnn.TemporalConvolution') require('cudnn.RNN') require('cudnn.RNNTanh') |