From ba5706c5d8a1a914a481e39ddb6bf05192ff9c89 Mon Sep 17 00:00:00 2001
From: Janek Mann <janekm@gmail.com>
Date: Sat, 21 Jan 2023 23:26:04 +0000
Subject: [PATCH] Fix wandb setup

Still need two have a log_writer
---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index 0883363..f71387f 100644
--- 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",
                                   )