From 100c7e55c1e120600a2e114e828a157d265a2411 Mon Sep 17 00:00:00 2001
From: WANG Yue <337111657@qq.com>
Date: Wed, 15 Sep 2021 21:27:16 +0800
Subject: [PATCH] Update run_gen.py

---
 run_gen.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/run_gen.py b/run_gen.py
index fd406b8..4e0a087 100644
--- a/run_gen.py
+++ b/run_gen.py
@@ -188,7 +188,7 @@ def main():
 
     if args.do_train:
         if args.local_rank in [-1, 0] and args.data_num == -1:
-            summary_fn = './tensorboard/{}'.format('/'.join(args.output_dir.split('/')[1:]))
+            summary_fn = '{}/{}'.format(args.summary_dir, '/'.join(args.output_dir.split('/')[1:]))
             tb_writer = SummaryWriter(summary_fn)
 
         # Prepare training data loader