e-lab/torch-toolbox

SpatialFullConvolution

varghesealex90 opened this issue · 1 comments

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

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