Based on discussions in Only batches of spatial targets supported (non-empty 3D tensors) but got targets of size: : [1, 1, 256, 256]. It might be due to the size of outputs or labels. Could you please print out their sizes?