fix(semseg): rename logger

Lightning-AI · Apr 16, 2020 · 138b616 · 138b616
1 parent 76a028b
commit 138b616
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pl_examples/domain_templates/semantic_segmentation.py b/pl_examples/domain_templates/semantic_segmentation.py
@@ -190,7 +190,7 @@ def main(hparams):
         logger = WandbLogger()
 
         # optional: log model topology
-        wandb_logger.watch(model.net)
+        logger.watch(model.net)
 
     # ------------------------
     # 3 INIT TRAINER