cudnn.torch
cudnn.torch copied to clipboard
SpatialBatchNormalization doesn't support affine=false
If affine
is set to false, the weight
and bias
are nil
, leading to an error in createIODescriptors. Maybe it would be better to assert in the constructor itself that affine==false
is not permitted at the moment ?