Add logger.

aistairc · Jan 28, 2025 · 6e4b9e0 · 6e4b9e0
1 parent 694f005
commit 6e4b9e0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/examples/torch/MNIST/config.yaml b/examples/torch/MNIST/config.yaml
@@ -1,4 +1,5 @@
 trainer:
+  logger: True
   max_epochs: 10
   callbacks:
     - _target_: lightning.pytorch.callbacks.ModelCheckpoint

diff --git a/examples/torch/MNIST/config_ddp.yaml b/examples/torch/MNIST/config_ddp.yaml
@@ -2,6 +2,7 @@ trainer:
   accelerator: "gpu"
   devices: 8
   strategy: "ddp"
+  logger: True
   max_epochs: 10
   callbacks:
     - _target_: lightning.pytorch.callbacks.ModelCheckpoint