Hi
I am getting this error RuntimeError: invalid argument 3: only batches of spatial targets supported (3D tensors) but got targets of dimension: 1 at /pytorch/aten/src/THNN/generic/SpatialClassNLLCriterion.c:61
whenever I used my saved model for transfer learning
Can anyone please help me out with this problem ?
Thanks in advance