I don’t know. when i print this it shows like this
Outputs tensor([[[[0.0454, 0.0453, 0.0452,  …, 0.0459, 0.0459, 0.0460],
[0.0455, 0.0454, 0.0452,  …, 0.0460, 0.0460, 0.0461],
[0.0457, 0.0455, 0.0452,  …, 0.0461, 0.0462, 0.0462],
…,
[0.0460, 0.0459, 0.0459,  …, 0.0457, 0.0457, 0.0458],
[0.0463, 0.0462, 0.0461,  …, 0.0461, 0.0462, 0.0463],
[0.0466, 0.0464, 0.0463,  …, 0.0464, 0.0466, 0.0468]]]],
device=‘cuda:0’, grad_fn=)
targets[:, 0] tensor([[[0, 0, 0,  …, 0, 0, 0],
[0, 0, 0,  …, 0, 0, 0],
[0, 0, 0,  …, 0, 0, 0],
…,
[0, 0, 0,  …, 0, 0, 0],
[0, 0, 0,  …, 0, 0, 0],
[0, 0, 0,  …, 0, 0, 0]]], device=‘cuda:0’)
Outputs Size torch.Size([1, 1, 256, 256])
targets[:, 0] Size torch.Size([1, 256, 256])