torch.Size([1104, 9, 64, 64]) torch.Size([1104, 1, 64, 64])
this is the data shape and the error is as follows:
RuntimeError: invalid argument 3: only batches of spatial targets supported (3D tensors) but got targets of dimension: 4 at C:\w\1\s\windows\pytorch\aten\src\THNN/generic/SpatialClassNLLCriterion.c:61