Are you working on a multi-class segmentation use case?
If so, then the target should have the shape [10, 256, 256] and contain values in the range [0, 5].
How did you end up with an output of [10,2048,256,256]?
It seems that the spacial size wasn’t reduced at all, which should be the case for a resnet.