Merge pull request #19 from janekm/wandb_fix

Fix wandb setup
2023-01-23 17:38:49 -08:00 · 2023-01-23 17:38:49 -08:00 · 4433bd7806
parent eea899f4a0 ba5706c5d8
commit 4433bd7806
1 changed files with 2 additions and 2 deletions
--- a/train.py
+++ b/train.py
@ -588,8 +588,8 @@ def main(args):

    if args.wandb is not None and args.wandb:
        wandb.init(project=args.project_name, sync_tensorboard=True, )
-    else:
-        log_writer = SummaryWriter(log_dir=log_folder, 
+  
+    log_writer = SummaryWriter(log_dir=log_folder, 
                                   flush_secs=5,
                                   comment="EveryDream2FineTunes",
                                  )