PMF
PMF copied to clipboard
only one element tensors can be converted to Python scalars
Thank you very much for your work, and I am getting errors like this during training:
only one element tensors can be converted to Python scalars
It can be trained normally at the beginning, but this error is reported after a few hundred iters. Have you ever met? HELP
Hmm, that's werid, can you print the value of loss_lov.item()
and input_feature.size(0)
.
A quickly suggestion is to comment this line, because it's just a logger of loss value, it won't effect the training process.
Best