Log SummaryWritters to Tensorboard

3 years ago · ab42b2856d
parent bfb29fb3be
commit ab42b2856d
1 changed files with 1 additions and 1 deletions
--- a/run_nerf.py
+++ b/run_nerf.py
@ -691,7 +691,7 @@ def train():
    print('VAL views are', i_val)

    # Summary writers
-    # writer = SummaryWriter(os.path.join(basedir, 'summaries', expname))
+    writer = SummaryWriter(os.path.join(basedir, 'summaries', expname))
    
    start = start + 1
    for i in trange(start, N_iters):