saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-07 00:13:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/orttraining/orttraining/python/training
History
zhijiang 1977b7ed6a
Fix pythonop training_mode in evaluation mode (#13514) 
Customer reported this issue: they see many warnings when doing hte
evaluation using ORTModule.


![image](https://user-images.githubusercontent.com/10530022/199371757-5fed7d05-a951-4f1b-8f88-049c5ab89886.png)

After investigation, we found the `training_mode` is exported to a wrong
value in evaluation mode, it's value should be 0, but we found it is 1.

Fix: 
fix pythonop training mode

if training_mode's type is torch._C._onnx.TrainingMode, then not matter
it is EVAL or TRAINING, "if training_mode" will always be true
2022-11-04 08:47:01 +08:00
..
amp Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
api Training API to export the eval model to an inference model (#13345) 2022-10-27 09:34:01 -07:00
experimental Make ORT callable from various Pytorch compilers (LazyTensor, TorchDynamo, etc) (#10460) 2022-08-22 09:40:40 -07:00
onnxblock Fix on-device training ExportModelForInferencing api (#13510) 2022-10-31 21:29:06 -07:00
optim [ORTModule] Update Supported DeepSpeed Version for FP16_Optimizer (#13305) 2022-10-13 13:03:01 +08:00
ortmodule Fix pythonop training_mode in evaluation mode (#13514) 2022-11-04 08:47:01 +08:00
torchdynamo Enable ORT in TorchDynamo (#13259) 2022-11-01 11:19:29 -07:00
utils/data Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
__init__.py [python] [orttraining] Add utility to export a graph to compute gradients (#8125) 2022-02-18 14:00:49 -08:00
_checkpoint_storage.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
_utils.py replace 'master' branch ref to 'main' for onnx repo (#12678) 2022-08-30 13:41:42 -07:00
checkpoint.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
model_desc_validation.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
orttrainer.py Deprecate ORTTrainer (#13022) 2022-09-23 18:10:09 -07:00
orttrainer_options.py Update ORTModule Default Opset Version to 15 (#12419) 2022-08-05 16:55:04 +08:00
postprocess.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00