UNet kernel_size=5 vs kernel_size=3

The increase in memory usage between a kernel size of 3 and 5 might be tiny as the majority of the memory usage would come from the intermediate forward activations, which might have the same shape.

1 Like