Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Billy Cao's avatar
    Add torch_empty_cache_steps to TrainingArguments (#31546) · 43ffb785
    Billy Cao authored 
    * Add torch_empty_cache_steps to TrainingArguments

* Fix formatting

* Add torch_empty_cache_steps to docs on single gpu training

* Remove check for torch_empty_cache_steps <= max_steps

* Captalize Tip

* Be device agnostic

* Fix linting
    43ffb785
training_args.py 150 KB