hoyeoplee/MeLU

enable anomaly detection to find the operation that failed to compute its gradient, with torch.autograd.set_detect_anomaly(True).

kbk12 opened this issue · 2 comments

kbk12 commented

enable anomaly detection to find the operation that failed to compute its gradient, with torch.autograd.set_detect_anomaly(True).

Sidsky commented

Hey, have you found any solution to fix this issue?

Sidsky commented

If the code fails to compile and throws the auto_grad error, then try changing the PyTorch version to 1.4 and use python 3.6