Operate under Inference Mode
as well as No Grad
#112
Labels
autograd
Tasks towards the online training / automatic differentiation feature
enhancement
New feature or request
hackathon
Hacktoberfest
Issues open for Hacktoberfest contributions
Originally commented by @ElliottKasoar as part of #103
No changes are currently included, but it would be good to support InferenceMode too eventually, as it should provide further performance benefits over NoGradMode.
However, it has stricter requirements, and the mode was only added (as a beta) in PyTorch 1.9, so we would need to be careful if we want to support older versions.
"Looking at the docs we'd maybe want to create an optional argument
bool :: inference mode
and if set to true enablec10::InferenceMode
for the function as a block.The text was updated successfully, but these errors were encountered: