Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

d

This commit is contained in:
awaelchli 2024-06-23 13:32:50 +02:00
parent baa5731984
commit aa6eb065b3
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/tests_pytorch/trainer/logging_/test_train_loop_logging.py
View File

@ -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")