HI i am also getting this error even though i have squeeze the labels but canot solve this problem
RuntimeError: Expected object of scalar type Long but got scalar type Float for argument #2 ‘target’ in call to _thnn_nll_loss2d_forward