diff options
author | Sam Gross <sgross@fb.com> | 2016-02-25 01:18:13 +0300 |
---|---|---|
committer | Sam Gross <sgross@fb.com> | 2016-02-25 22:50:15 +0300 |
commit | c327f5b86ad7972f12fd5fe41c7215eea381043d (patch) | |
tree | 0edeaee6e9e069b069ae4feddd44dea60c1accaa /init.lua | |
parent | 00a60967687741ffdac20a8dda924127f8ccae4e (diff) |
Add VolumetricBatchNormalization
The BatchNormalization modules now all extend nn.BatchNormalization and
use the same THNN/THCUNN implementation.
Diffstat (limited to 'init.lua')
-rw-r--r-- | init.lua | 1 |
1 files changed, 1 insertions, 0 deletions
@@ -113,6 +113,7 @@ require('nn.VolumetricFullConvolution') require('nn.VolumetricMaxPooling') require('nn.VolumetricMaxUnpooling') require('nn.VolumetricAveragePooling') +require('nn.VolumetricBatchNormalization') require('nn.ParallelTable') require('nn.ConcatTable') |