Solved by ptrblck in post #3 
              
              
                Usually this type of error is raised if you have some skip connections in the model (or anything similar where you are concatenating activations) and are using an “unsupported” spatial shape for the input. 
“Unsupported” means that the model wasn’t designed to work with any arbitrary shape, but ofte…