Wandb saves model checkpoints

JonathanCrabbe · Dec 27, 2023 · b805018 · b805018
1 parent c91794a
commit b805018
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cmd/conf/trainer/default.yaml b/cmd/conf/trainer/default.yaml
@@ -5,5 +5,6 @@ gradient_clip_val: 1.0
 enable_progress_bar: true
 logger:
   _target_: pytorch_lightning.loggers.WandbLogger
+  log_model: true
 defaults:
   - callbacks: default