Use grad_transform in torch.autograd.Function lookaside
#4555
| Job | Run time |
|---|---|
| 12m 10s | |
| 23m 51s | |
| 13m 8s | |
| 23m 45s | |
| 7m 42s | |
| 17m 27s | |
| 9m 2s | |
| 18m 59s | |
| 21m 46s | |
| 20m 40s | |
| 20m 24s | |
| 25m 14s | |
| 1s | |
| 3h 34m 9s |