Fix wandb setup

Still need two have a log_writer
2023-01-21 23:26:04 +00:00 · 2023-01-21 23:26:04 +00:00 · ba5706c5d8
parent 1c2708dc63
commit ba5706c5d8
1 changed files with 2 additions and 2 deletions
--- a/train.py
+++ b/train.py
@ -577,8 +577,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",
                                  )