saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-01 23:30:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/orttraining/orttraining/python/training
History
Thiago Crepaldi d4917f2d65 Hard-code input types for DropoutGrad on BERT
2020-12-15 09:03:08 -08:00
..
amp Move onnxruntime.experiment to onnxruntime.training namespace (#5045) 2020-09-09 09:46:06 -07:00
optim Move onnxruntime.experiment to onnxruntime.training namespace (#5045) 2020-09-09 09:46:06 -07:00
__init__.py Forward pass using InferenceSession on exported ONNX 2020-12-15 09:03:07 -08:00
_checkpoint_storage.py save data to and load data from an hdf5 file for checkpointing (#5975) 2020-12-08 11:40:57 -08:00
_utils.py state_dict and load_state_dict for ORTTrainer (#6095) 2020-12-14 11:55:52 -08:00
checkpoint.py Enable passing initial optimizer state while creating training session (#5869) 2020-12-08 21:20:51 -05:00
model_desc_validation.py Move onnxruntime.experiment to onnxruntime.training namespace (#5045) 2020-09-09 09:46:06 -07:00
ortmodule.py Hard-code input types for DropoutGrad on BERT 2020-12-15 09:03:08 -08:00
orttrainer.py Add script to run Flexible API MVP PoC 2020-12-15 09:03:07 -08:00
orttrainer_options.py Add script to run Flexible API MVP PoC 2020-12-15 09:03:07 -08:00
postprocess.py Move onnxruntime.experiment to onnxruntime.training namespace (#5045) 2020-09-09 09:46:06 -07:00