From aa6eb065b3396b07e8ea936992ac2ce7aa5459bf Mon Sep 17 00:00:00 2001 From: awaelchli Date: Sun, 23 Jun 2024 13:32:50 +0200 Subject: [PATCH] d --- .../trainer/logging_/test_train_loop_logging.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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")