Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix #997 (#1018)

This commit is contained in:
Bilal Khan 2020-03-02 20:51:05 -06:00 committed by GitHub
parent 6dae5698ef
commit 29cbc9e723
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/trainer/distrib_parts.py
View File

@ -489,7 +489,7 @@ class TrainerDPMixin(ABC):
# init 16 bit for TPU # init 16 bit for TPU
if self.precision == 16: 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}, ' \ m = f'INIT TPU local core: {self.tpu_local_core_rank}, ' \
f'global rank: {self.tpu_global_core_rank}' f'global rank: {self.tpu_global_core_rank}'