Fix TF training arguments instantiation (#8063)

3a107645 · Lysandre Debut · GitHub · bc9332b5 · 3a107645
Unverified Commit 3a107645 authored Oct 26, 2020 by Lysandre Debut Committed by GitHub Oct 26, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/training_args.py src/transformers/training_args.py +1 -1

No files found.
--- a/src/transformers/training_args.py
+++ b/src/transformers/training_args.py
@@ -351,7 +351,7 @@ class TrainingArguments:
        if self.run_name is None:
            self.run_name = self.output_dir

-        if self.device.type != "cuda" and self.fp16:
+        if is_torch_available() and self.device.type != "cuda" and self.fp16:
            raise ValueError("AMP (`--fp16`) can only be used on CUDA devices.")

    @property