microsoft / knossos-ksc

Compiler with automatic differentiation
Other
45 stars 10 forks source link

Benchmark: use no_grad() in PyTorch for inference #895

Closed cgravill closed 3 years ago

cgravill commented 3 years ago

Using longer runs (5 seconds) from #894

Before: image

After: image

It's possible this helped inference but note there are some overall shifts on forward and backwards as well. Given this is the right thing to do I suggest adding this in any case.