transformers
Delete `state_dict` to release memory as early as possible
#18832

Merged

Delete `state_dict` to release memory as early as possible #18832

ydshieh merged 4 commits into main from delete_state_dict_to_release_mem

ydshieh requested a review from

LysandreJik 4 years ago

ydshieh requested a review from

sgugger 4 years ago

ydshieh marked this pull request as draft 4 years ago

del a copy of state_dict to release memory as early as possible

50f7efde

A change in order to pass black check

52f23883

fix missing arg

45db5e69

ydshieh force pushed to 45db5e69 4 years ago

ydshieh marked this pull request as ready for review 4 years ago

sgugger approved these changes on 2022-08-31

LysandreJik approved these changes on 2022-09-01

Update src/transformers/modeling_utils.py

3dcb2b7b

ydshieh merged 563a8d58 into main 4 years ago

ydshieh deleted the delete_state_dict_to_release_mem branch 4 years ago

Reviewers

LysandreJik

sgugger

Assignees

No one assigned

Labels

None yet

Milestone

No milestone