wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-28 13:20:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix #997 (#1018)

Browse Source
This commit is contained in:
Bilal Khan authored and GitHub committed 2020-03-02 21:51:05 -05:00
1 parent 6dae5698ef
commit 29cbc9e723
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/distrib_parts.py
+1 -1
View File
@@ -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}'