From 6e4b9e063202dce26cd9200927a1f8554855b0b7 Mon Sep 17 00:00:00 2001 From: KanaiYuma Date: Tue, 28 Jan 2025 16:54:20 +0900 Subject: [PATCH] Add logger. --- examples/torch/MNIST/config.yaml | 1 + examples/torch/MNIST/config_ddp.yaml | 1 + 2 files changed, 2 insertions(+) diff --git a/examples/torch/MNIST/config.yaml b/examples/torch/MNIST/config.yaml index 41c45f19..811de63d 100644 --- 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 index a7061c2d..b42b3798 100644 --- 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