otthorn/nerf-pytorch
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Log SummaryWritters to Tensorboard

Browse source
This commit is contained in:
otthorn 2021-01-22 12:25:09 +01:00
parent bfb29fb3be
commit ab42b2856d
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
run_nerf.py
View file

@ -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):