PR #6095 state_dict and load_state_dict for ORTTrainer

state_dict and load_state_dict for ORTTrainer #6095

baijumeswani merged 5 commits into master from bmeswani/state_dict

baijumeswani requested a review from

BowenBao 5 years ago

baijumeswani requested a review from

liqunfu 5 years ago

baijumeswani requested a review from

spandantiwari 5 years ago

baijumeswani requested a review from

thiagocrepaldi 5 years ago

baijumeswani requested a review 5 years ago

baijumeswani force pushed 5 years ago

baijumeswani force pushed to 22e698a9 5 years ago

ashbhandare commented on 2020-12-10

add functions state_dict and load_state_dict to ORTTrainer

16013eb4

unit tests for state_dict and load_state_dict for ORTTrainer

6b28613c

baijumeswani force pushed from 22e698a9 to 6b28613c 5 years ago

ashbhandare requested changes on 2020-12-10

thiagocrepaldi requested changes on 2020-12-10

name changes, docstrings, utility functions

4043870f

baijumeswani force pushed to 4043870f 5 years ago

datatype to numpy array when pytorch_format=False, strict checking at…

798aeb79

baijumeswani force pushed to 798aeb79 5 years ago

ashbhandare commented on 2020-12-11

return all frozen weights in the fp32 bucket, warn when pytorch_forma…

b42caea2

ashbhandare approved these changes on 2020-12-14

thiagocrepaldi approved these changes on 2020-12-14

baijumeswani merged dd2e5a1a into master 5 years ago

baijumeswani deleted the bmeswani/state_dict branch 5 years ago

Reviewers

thiagocrepaldi

ashbhandare

BowenBao

liqunfu

spandantiwari

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

onnxruntime state_dict and load_state_dict for ORTTrainer #6095 Merged

state_dict and load_state_dict for ORTTrainer #6095

onnxruntime
state_dict and load_state_dict for ORTTrainer
#6095

Merged