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

[kbk12]

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

[Sidsky]

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

[Sidsky]

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