Remove LighterLogger fom the CIFAR10 yaml

project-lighter · Jan 16, 2024 · d55ed81 · d55ed81
1 parent 4fa5dc7
commit d55ed81
Showing 1 changed file with 5 additions and 11 deletions.
diff --git a/projects/cifar10/experiments/monai_bundle_prototype.yaml b/projects/cifar10/experiments/monai_bundle_prototype.yaml
@@ -7,18 +7,12 @@ trainer:
     # devices: 1  # 2
     # strategy: ddp
     log_every_n_steps: 10
-    logger: False
-    callbacks:
-        - _target_: lighter.callbacks.LighterLogger
-          project: CIFAR10
-          log_dir: "$@project + '/logs' "
-          tensorboard: True
-          wandb: True
-          input_type: image
-          # pred_type: scalar
-          # target_type: scalar
-          max_samples: 10
+    logger:
+        _target_: pytorch_lightning.loggers.WandbLogger
+        project: CIFAR10
+        save_dir: "$@project + '/logs'"
 
+    callbacks:
         - _target_: lighter.callbacks.LighterFileWriter
           directory: "$@project + '/predictions' "
           format: "tensor"