"src/vscode:/vscode.git/clone" did not exist on "26aa24f8356c7f96180f81061d97a985f1152815"
Unverified Commit 0b9c9345 authored by Quentin Gallouédec's avatar Quentin Gallouédec Committed by GitHub
Browse files

Enforce pin memory disabling when using cpu only (#27745)

if use_cpu: dataloader_pin_memory = False
parent fdd86eed
......@@ -1734,6 +1734,9 @@ class TrainingArguments:
self.deepspeed_plugin.set_mixed_precision(mixed_precision)
self.deepspeed_plugin.set_deepspeed_weakref()
if self.use_cpu:
self.dataloader_pin_memory = False
if self.push_to_hub_token is not None:
warnings.warn(
"`--push_to_hub_token` is deprecated and will be removed in version 5 of 🤗 Transformers. Use "
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment