inference_mode
class torch.inference_mode(mode=True)
Context-manager that enables or disables inference mode
InferenceMode is a new context manager analogous to "no_grad" to be
   used when you are certain your operations will have no interactions
   with autograd (e.g., model training). Code run under this mode gets
   better performance by disabling view tracking and version counter
   bumps. Note that unlike some other mechanisms that locally enable
   or disable grad, entering inference_mode also disables to forward-
   mode AD.
This context manager is thread local; it will not affect
   computation in other threads.
Also functions as a decorator. (Make sure to instantiate with
   parenthesis.)
Note:
 Inference mode is one of several mechanisms that can enable or
 disable gradients locally see Locally disabling gradient
 computation for more information on how they compare.

Parameters: