train on pytorch1.5
xingaoli opened this issue · 0 comments
xingaoli commented
It is all right when I only switch the torch version to 1.1. But I find the loss is nan after 20 iters when I try to train the model on torch1.5. I confirm that only the PyTorch version is different. Do you know what this is about?