-
Notifications
You must be signed in to change notification settings - Fork 19
/
tensorboard_demo.py
27 lines (24 loc) · 867 Bytes
/
tensorboard_demo.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
import torch
from tensorboardX import SummaryWriter
import time
writer = SummaryWriter('train_log')
x = torch.FloatTensor([100])
y = torch.FloatTensor([500])
for epoch in range(100):
x /= 1.5
y /= 1.5
loss = y - x
print(loss)
writer.add_histogram('zz/x', x, epoch)
writer.add_histogram('zz/y', y, epoch)
writer.add_scalar('data/x', x, epoch)
writer.add_scalar('data/y', y, epoch)
writer.add_scalar('data/loss', loss, epoch)
writer.add_scalars('data/scalar_group', {'x': x,
'y': y,
'loss': loss}, epoch)
writer.add_text('zz/text', 'zz: this is epoch ' + str(epoch), epoch)
time.sleep(0.5)
# export scalar data to JSON for external processing
writer.export_scalars_to_json("./test.json")
writer.close()