The error points to a shape mismatch in the model output and target.
It seems one of these tensors has a batch size of 144 while the other has only 4 samples.
Print the shapes of the output and target before passing them to the criterion and make sure they have the same number of samples.