The shape mismatch error is raised in the torch.cat operation since x and skip have different spatial sizes. Pad one of these tensors or slice the other one to create matching spatial sizes.
1 Like
The shape mismatch error is raised in the torch.cat operation since x and skip have different spatial sizes. Pad one of these tensors or slice the other one to create matching spatial sizes.