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

Merged

Adding support for saving and loading train step info properties in the state dict and checkpoint file. #10569

mpapdiwala merged 8 commits into microsoft:master from mpapdiwala:dev/mpapdiwala/lr-scheduler

Adding optimization step and step parameter to the ORTTrainer constru…

a17f3600

mpapdiwala requested a review from

thiagocrepaldi 3 years ago

mpapdiwala requested a review from

tlh20 3 years ago

mpapdiwala requested a review from

liqunfu 3 years ago

mpapdiwala requested a review from

baijumeswani 3 years ago

mpapdiwala requested a review from

SherlockNoMad 3 years ago

mpapdiwala requested a review from

xadupre 3 years ago

pallavides requested a review from

souptc 3 years ago

mpapdiwala closed this 3 years ago

mpapdiwala reopened this 3 years ago

thiagocrepaldi requested changes on 2022-02-22

Added ORTTrainerOptions for optimization step

931764ec

Adding Train Step Info Settings to State Dictionary

bba7792b

Adding train step info key

39cbe2ba

Updating comments

caf6d5a6

Reverting changes

8e0a314e

Merge branch 'lr-scheduler-draft' into dev/mpapdiwala/lr-scheduler

a8152120

Updating test case for new state dict entry train_step_info

6912caac

souptc approved these changes on 2022-03-21

baijumeswani approved these changes on 2022-03-22

mpapdiwala changed the title ~~Adding optimization step parameters to the ORTTrainer constructor~~ Adding support for saving and loading train step info properties in the state dict and checkpoint file. 3 years ago

thiagocrepaldi approved these changes on 2022-03-24

mpapdiwala merged 1e917c87 into master 3 years ago

mpapdiwala deleted the dev/mpapdiwala/lr-scheduler branch 3 years ago

Reviewers

baijumeswani

souptc

thiagocrepaldi

tlh20

liqunfu

SherlockNoMad

xadupre

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

onnxruntime Adding support for saving and loading train step info properties in the state dict and checkpoint file. #10569 Merged

Adding support for saving and loading train step info properties in the state dict and checkpoint file. #10569

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

Merged