search

e-lab / torch-toolbox

A collection of snippets and libraries for Torch from e-Lab
https://engineering.purdue.edu/elab/
199 stars 64 forks source link

SpatialFullConvolution #23

Open varghesealex90 opened 7 years ago

varghesealex90 commented 7 years ago

Hi, Is there xavier initilzation for SpatialFullConvolution? is it sufficient to add the following line in your code? elseif m.__typename == 'nn.SpatialFullConvolution' then m:reset(method(m.nInputPlanem.kHm.kW, m.nOutputPlanem.kHm.kW))

jhjin commented 7 years ago

Hey, @varghesealex90 you are always welcome to make a PR!