DiffICM / 1_feature_extractor /test_board.py
Qiyp's picture
code of stage1 & 3, remove large files
1633fcc
from torch.utils.tensorboard import SummaryWriter
import numpy as np
np.random.seed(20200910)
writer = SummaryWriter()
for n_iter in range(100):
writer.add_scalar('Loss/train', np.random.random(), n_iter)
writer.add_scalar('Loss/test', np.random.random(), n_iter)
writer.add_scalar('Accuracy/train', np.random.random(), n_iter)
writer.add_scalar('Accuracy/test', np.random.random(), n_iter)