f

J13253902839 · Aug 18, 2020 · 90827c9 · 90827c9
1 parent 67d7eb7
commit 90827c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -15,7 +15,7 @@ def main(cfg):
     if cfg.wandb.project:
         import wandb
         from wandb.keras import WandbCallback
-        wandb.init(project="age-gender-estimation")
+        wandb.init(project=cfg.wandb.project)
         callbacks = [WandbCallback()]
     else:
         callbacks = []