RuntimeError: 1only batches of spatial targets supported (3D tensors) but got targets of size: : [1, 3, 96, 128]

vision
4

No, I meant you would have to convert the colors to class indices.
If your dataset doesn’t provide the mapping, you could create your own as described here.

1 Like