From 446a1e23d7fe3b2e07f1a5887fe819d0dfa7d4e0 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 22 Feb 2020 05:15:36 -0500 Subject: [PATCH] Update training_loop.py (#913) --- pytorch_lightning/trainer/training_loop.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/training_loop.py b/pytorch_lightning/trainer/training_loop.py index e59183aa6b..87626d8cdb 100644 --- a/pytorch_lightning/trainer/training_loop.py +++ b/pytorch_lightning/trainer/training_loop.py @@ -594,7 +594,7 @@ class TrainerTrainLoopMixin(ABC): model = self.get_model() with self.profiler.profile('optimizer_step'): if self.use_tpu: - xm.optimizer_step(optimizer, barrier=True) + xm.optimizer_step(optimizer) else: model.optimizer_step(self.current_epoch, batch_idx, optimizer, opt_idx, optimizer_closure)