diff --git a/tests/tests_pytorch/trainer/logging_/test_train_loop_logging.py b/tests/tests_pytorch/trainer/logging_/test_train_loop_logging.py index 8f905c9a24..068ca71505 100644 --- a/tests/tests_pytorch/trainer/logging_/test_train_loop_logging.py +++ b/tests/tests_pytorch/trainer/logging_/test_train_loop_logging.py @@ -322,10 +322,10 @@ class LoggingSyncDistModel(BoringModel): def training_step(self, batch, batch_idx): value = self.fake_result + self.rank self.log("foo", value, on_step=True, on_epoch=False, sync_dist=True, reduce_fx="sum") - # self.log("foo_2", 2, on_step=True, on_epoch=False, sync_dist=True, reduce_fx="sum") - # self.log("foo_3", 2, on_step=True, on_epoch=False, sync_dist=True, reduce_fx="mean") - # self.log("foo_4", value, on_step=True, on_epoch=False, sync_dist=True, reduce_fx="mean") - # self.log("foo_5", batch_idx + self.rank, on_step=True, on_epoch=False, sync_dist=True, reduce_fx="max") + self.log("foo_2", 2, on_step=True, on_epoch=False, sync_dist=True, reduce_fx="sum") + self.log("foo_3", 2, on_step=True, on_epoch=False, sync_dist=True, reduce_fx="mean") + self.log("foo_4", value, on_step=True, on_epoch=False, sync_dist=True, reduce_fx="mean") + self.log("foo_5", batch_idx + self.rank, on_step=True, on_epoch=False, sync_dist=True, reduce_fx="max") # # self.log("foo_6", value, on_step=False, on_epoch=True, sync_dist=True, reduce_fx="sum") # self.log("foo_7", 2, on_step=False, on_epoch=True, sync_dist=True, reduce_fx="sum")