We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
torch.no_grad
It would be nice for TorchFix to flag when no_grad is used both as a decorator and a context manager in the same function:
no_grad
import torch @torch.no_grad def some_func(): with torch.no_grad(): # Redundant to decorator ...
Same thing applies for torch.inference_mode.
torch.inference_mode