From 29cbc9e7230dd4c5e5e8e8ff789b838ed4a79e20 Mon Sep 17 00:00:00 2001 From: Bilal Khan Date: Mon, 2 Mar 2020 20:51:05 -0600 Subject: [PATCH] Fix #997 (#1018) --- pytorch_lightning/trainer/distrib_parts.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/distrib_parts.py b/pytorch_lightning/trainer/distrib_parts.py index 440afc06..286e71ef 100644 --- a/pytorch_lightning/trainer/distrib_parts.py +++ b/pytorch_lightning/trainer/distrib_parts.py @@ -489,7 +489,7 @@ class TrainerDPMixin(ABC): # init 16 bit for TPU if self.precision == 16: - os.environ['XLA_USE_BF16'] = 1 + os.environ['XLA_USE_BF16'] = str(1) m = f'INIT TPU local core: {self.tpu_local_core_rank}, ' \ f'global rank: {self.tpu_global_core_rank}'