deactivate autograd during validation

aiqm / torchani

Accurate Neural Network Potential on PyTorch

https://aiqm.github.io/torchani/

MIT License

459 stars 126 forks source link

Closed IgnacioJPickering closed 3 years ago

IgnacioJPickering commented 3 years ago

Deactivate autograd engine for validation examples and add model.train(False) and model.train(True) for future insurance