PR #15803 Refine a bit (on device training)

Refine a bit (on device training) #15803

askhade merged 3 commits into main from pengwa/cleanup

simplify ParameterOptimizerState a bit && use inlined containers

48de508e

remove GetStateDict APIs

3d6a7635

re-enable cuda test for lr scheduler

387e8eea

pengwa requested a review from

askhade 2 years ago

pengwa requested a review from

baijumeswani 2 years ago

pengwa added training

baijumeswani approved these changes on 2023-05-10

askhade commented on 2023-05-11

askhade approved these changes on 2023-05-11

askhade merged fed52053 into main 2 years ago

askhade deleted the pengwa/cleanup branch 2 years ago

Reviewers

askhade

baijumeswani

Assignees

No one assigned

Labels

training

Milestone

No milestone