Log SummaryWritters to Tensorboard
This commit is contained in:
parent
bfb29fb3be
commit
ab42b2856d
1 changed files with 1 additions and 1 deletions
|
@ -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):
|
||||||
|
|
Loading…
Reference in a new issue