Update distrib_parts.py

Lightning-AI · Jul 11, 2020 · 89c27d4 · 89c27d4
1 parent 5dce449
commit 89c27d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/trainer/distrib_parts.py b/pytorch_lightning/trainer/distrib_parts.py
@@ -192,7 +192,7 @@ def tpu_train(self, tpu_core_idx, model):
             model.prepare_data()
             self._is_data_prepared = True
 
-        self.barrier()
+        self.barrier('prepare_data')
 
         # call setup after the ddp process has connected
         self.setup('fit')