Open Philip-PKU opened 2 years ago
Go to functions.py and add the line below at the top after the imports.
torch.autograd.set_detect_anomaly(True)
Also refer to this thread if you encounter one more error after executing the previous step. #108
select pytorch version of 1.6 (maybe!)