【转载】轻松学Pytorch-Pytorch可视化
import numpy as np
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter(log_dir='./log')
flag = 0
if flag :
for x in range(100):
# 把x*2的数据加入标签y=2x的曲线
writer.add_scalar(tag='y=2x',scalar_value=x*2,global_s..
转载
2021-04-21 11:39:04 ·
214 阅读 ·
0 评论