diff options
author | Soumith Chintala <soumith@gmail.com> | 2016-10-17 02:00:46 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2016-10-17 02:00:46 +0300 |
commit | 7c17669434bda5741b108e023fe37fa7d7854fef (patch) | |
tree | 0e970bebeb61b10a52fd4708277f4c017ee09c2c /init.lua | |
parent | a79f22e9e35caca0a01465dd4f378a3ba4d127bc (diff) | |
parent | 42619d161e45413967c6e107f0ed14857626e494 (diff) |
Merge pull request #1005 from alykhantejani/add_pixel_shuffle_layer
PixelShuffle module for sub-pixel convolution
Diffstat (limited to 'init.lua')
-rw-r--r-- | init.lua | 2 |
1 files changed, 2 insertions, 0 deletions
@@ -175,6 +175,8 @@ require('nn.BCECriterion') require('nn.CrossEntropyCriterion') require('nn.ParallelCriterion') +require('nn.PixelShuffle') + require('nn.StochasticGradient') require('nn.MM') |