added wandb

czyjtu · May 1, 2023 · 097c888 · 097c888
1 parent c7c36e6
commit 097c888
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -132,3  132,4 @@ dmypy.json
 output
 outputs
 lightning_logs
 wandb
diff --git a/main-ml.py b/main-ml.py
@@ -26,7  26,7 @@ def train(cfg: DictConfig, datamodule: pl.LightningDataModule, task: pl.Lightnin
         dirpath="output/",
         filename=f"cad-{cfg.model.name}-{{epoch:03d}}-{{f1:.4f}}",
     )
-    # wandb_logger = WandbLogger(project="???")
     wandb_logger = WandbLogger(project="cad-ml")
     trainer = pl.Trainer(
         accelerator=cfg.task.accelerator,
         # logger=wandb_logger,
-Original file line number
+Diff line change
@@ Expand Up / @@ -132,3 132,4 @@ dmypy.json @@
     output
     outputs
     lightning_logs
+    wandb