diff --git a/cmd/conf/trainer/default.yaml b/cmd/conf/trainer/default.yaml index 8d4dd38..974c127 100644 --- 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