From 3a10764574f252591eeaa5bbb10b778f623a4814 Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Mon, 26 Oct 2020 18:39:25 +0000 Subject: [PATCH] Fix TF training arguments instantiation (#8063) --- src/transformers/training_args.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/training_args.py b/src/transformers/training_args.py index b86a1cbc2b..0aa06902e3 100644 --- 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