onnxruntime
1e917c87 - Adding support for saving and loading train step info properties in the state dict and checkpoint file. (#10569)

Commit
3 years ago
Adding support for saving and loading train step info properties in the state dict and checkpoint file. (#10569) * Adding optimization step and step parameter to the ORTTrainer constructor * Added ORTTrainerOptions for optimization step * Adding Train Step Info Settings to State Dictionary * Adding train step info key * Updating comments * Reverting changes * Updating test case for new state dict entry train_step_info
Author
Parents
Loading