PR #27745 Enforce pin memory disabling when using cpu only

Enforce pin memory disabling when using cpu only #27745

ArthurZucker merged 1 commit into huggingface:main from qgallouedec:fix-use-cpu-pin-memory

if use_cpu: dataloader_pin_memory = False

0df52e7e

muellerzr approved these changes on 2023-11-28

muellerzr requested a review from

ArthurZucker 2 years ago

qgallouedec changed the title ~~Fix training failure when GPU memory saturated and `--use_cpu`~~ Enforce pin memory disabling when using cpu only 2 years ago

ArthurZucker approved these changes on 2023-11-28

ArthurZucker merged 0b9c9345 into main 2 years ago

qgallouedec deleted the fix-use-cpu-pin-memory branch 2 years ago

Reviewers

ArthurZucker

muellerzr

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

transformers Enforce pin memory disabling when using cpu only #27745 Merged

Enforce pin memory disabling when using cpu only #27745

transformers
Enforce pin memory disabling when using cpu only
#27745

Merged