Update distrib_parts.py

Lightning-AI · Aug 11, 2020 · e23ffbc · e23ffbc
1 parent 2ba7d73
commit e23ffbc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/accelerators/tpu_backend.py b/pytorch_lightning/accelerators/tpu_backend.py
@@ -79,7 +79,7 @@ def train(self):
             self.trainer.model.prepare_data()
             self._is_data_prepared = True
 
-        self.trainer.barrier()
+        self.trainer.barrier('prepare_data')
 
         # train
         if self.trainer.tpu_id is not None: