From e23ffbc901aee984cf92002bd67c124d3c86340c Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 11 Jul 2020 18:43:49 -0400 Subject: [PATCH] Update distrib_parts.py --- pytorch_lightning/accelerators/tpu_backend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/accelerators/tpu_backend.py b/pytorch_lightning/accelerators/tpu_backend.py index 6e82205f9c3da..0aafc761ea0cc 100644 --- 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: