Commits


mpapdiwala authored and GitHub committed 1e917c879e3
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