Because of the way sizes are used in Cascade, there is an error thrown if you don't get it correct. If you change the multiple of the width and height to step in units of 128, instead of 8, then this bypasses the errors.
Example error:
RuntimeError: pixel_unshuffle expects height to be divisible by downscale_factor, but input.size(-2)=277 is not divisible by 2
Because of the way sizes are used in Cascade, there is an error thrown if you don't get it correct. If you change the multiple of the width and height to step in units of 128, instead of 8, then this bypasses the errors.
Example error:
RuntimeError: pixel_unshuffle expects height to be divisible by downscale_factor, but input.size(-2)=277 is not divisible by 2