torchmore
torchmore copied to clipboard
Padding added to cater for the values dropped during downsampling in U-Net
Assertion removed which assumes the size of the feature maps to be even. Added padding instead to cater to these cases since having a feature volume of 7x7 is quite common which cannot be handled by the current implementation.