pytorch
c8b9b626 - [ONNX] Fix arg type in `_set_training_mode` (#78583)

Commit
2 years ago
[ONNX] Fix arg type in `_set_training_mode` (#78583) When `TrainingMode.PRESERVE` is set for export, the exporter used to change the model's training mode based on some logic. Now we respect the option and not touch the model's training state. - Previously `_set_training_mode`'s behavior doesn't match what the global variable expects. This PR removes the deprecated `_set_training_mode` and makes the type correct. Pull Request resolved: https://github.com/pytorch/pytorch/pull/78583 Approved by: https://github.com/BowenBao
Author
Committer
Parents
Loading