PR #31546 Add torch_empty_cache_steps to TrainingArguments

Add torch_empty_cache_steps to TrainingArguments #31546

muellerzr merged 8 commits into huggingface:main from aliencaocao:empty-cache-arg

Add torch_empty_cache_steps to TrainingArguments

b20828da

Fix formatting

ee041092

Add torch_empty_cache_steps to docs on single gpu training

32d0c04d

amyeroberts commented on 2024-06-23

Remove check for torch_empty_cache_steps <= max_steps

e9299ba5

Merge branch 'refs/heads/main' into empty-cache-arg

d4650989

muellerzr approved these changes on 2024-07-01

Captalize Tip

c24bd3da

amyeroberts approved these changes on 2024-07-02

muellerzr requested changes on 2024-07-04

Be device agnostic

ca4107cf

muellerzr approved these changes on 2024-07-04

Fix linting

fcae6266

muellerzr merged 43ffb785 into main 1 year ago

aliencaocao deleted the empty-cache-arg branch 1 year ago

Reviewers

muellerzr

amyeroberts

Assignees

No one assigned

Labels

None yet

Milestone

No milestone