ptrblck:

, you could of course remove it. Since the output shape is defined in this layer, you might need to use F.interpolate or another method to create the same output shape.


Sure… thank you for your input