Log SummaryWritters to Tensorboard

This commit is contained in:
otthorn 2021-01-22 12:25:09 +01:00
parent bfb29fb3be
commit ab42b2856d

View file

@ -691,7 +691,7 @@ def train():
print('VAL views are', i_val) print('VAL views are', i_val)
# Summary writers # Summary writers
# writer = SummaryWriter(os.path.join(basedir, 'summaries', expname)) writer = SummaryWriter(os.path.join(basedir, 'summaries', expname))
start = start + 1 start = start + 1
for i in trange(start, N_iters): for i in trange(start, N_iters):