Update distrib_parts.py

Lightning-AI · Sep 7, 2020 · 0cfcb8a · 0cfcb8a
1 parent 7c7b7cc
commit 0cfcb8a
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
@@ -86,7 +86,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: