Don't print configs in tensorboard

shinianzhihou · Mar 24, 2020 · d65d882 · d65d882
1 parent 1933d3b
commit d65d882
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 25 deletions.
diff --git a/build/tensorboard.py b/build/tensorboard.py
@@ -10,6 +10,6 @@ def build_tensorboad(cfg):
     name = "%s/%s" % ( time.strftime("%Y-%m-%d-%H-%M"),ID)
     path = os.path.join(tcfg.PATH, name)
     writer = SummaryWriter(path)
-    writer.add_text("config", str(cfg), 0)
+    # writer.add_text("config", str(cfg), 0)
     return writer
 
diff --git a/engine/trainer.py b/engine/trainer.py
@@ -50,7 +50,6 @@ def train_epoch(
             writer.add_scalar("train/lr", scheduler.get_lr()[0], current_batch)
 
 
-
         if test_loader and not current_batch % cfg.SOLVER.TEST_PERIOD:
             step = current_batch // cfg.SOLVER.TEST_PERIOD
             del img1,img2,gt,output

diff --git a/logs/eval.csv b/logs/eval.csv