Update test_trainer.py

Lightning-AI · Apr 7, 2021 · 5d83193 · 5d83193
1 parent 27db048
commit 5d83193
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py
@@ -1906,7 +1906,7 @@ def validation_step(self, batch, batch_idx):
 
         def validation_epoch_end(self, outputs) -> None:
             avg_x = torch.stack([x['x'] for x in outputs]).mean()
-            self.log('avg_x', avg_x)
+            self.log('avg_x', avg_x, sync_dist=True)
 
     model = TestModel()
     checkpoint = ModelCheckpoint(dirpath=tmpdir, monitor='avg_x', mode='min', save_top_k=1)