At trainer.py, line 384>
loss_m = criterion(torch.log(output.clamp(min=1e-8)), proj_labels) + self.ls(output, proj_labels.long())
should be replaced with
loss_m = criterion(torch.log(output.clamp(min=1e-8)), proj_labels.long()) + self.ls(output, proj_labels.long())
At trainer.py, line 384>
loss_m = criterion(torch.log(output.clamp(min=1e-8)), proj_labels) + self.ls(output, proj_labels.long())
should be replaced withloss_m = criterion(torch.log(output.clamp(min=1e-8)), proj_labels
.long()) + self.ls(output, proj_labels.long())