chore(optim): wrap torch.autograd.grad() with torch.enable_grad() context
#1695
| Job | Run time |
|---|---|
| 21m 15s | |
| 17m 5s | |
| 18m 50s | |
| 28m 0s | |
| 1h 25m 10s |
torch.autograd.grad() with torch.enable_grad() context
#1695
| Job | Run time |
|---|---|
| 21m 15s | |
| 17m 5s | |
| 18m 50s | |
| 28m 0s | |
| 1h 25m 10s |