Only batches of spatial targets supported (non-empty 3D tensors) but got targets of size: : [1, 1, 256, 256]

48

Ah okay, this is so helpful. Thank you so much @ptrblck!!