Thomas Tschinkel

icon location Italy icon location 15 y/o | learning AI, ML & Deep Learning 🤖📚

Results 1 issues of Thomas Tschinkel

Use of torch.inference_mode() instead of torch.no_grad() for inference

3 comment

Replaced the `torch.no_grad()` context manager with the `torch.inference_mode()` context manager in the test function. Why? `torch.inference_mode()` is more efficient as it reduces memory usage, inference speed and it disabled autograd...

cla signed