(1)导入tensorboard,要先实例化SummaryWriter类,指明记录日志的路径信息:
from torch.utils.tensorboard import SummaryWriter
(2)先实例化SummaryWriter类,指明记录日志的路径信息:
writer = SummaryWriter(log_dir='路径地址')
(3)调用相应的API接口,一般格式:
add_xxx(标签名,对象,迭代次数)
(4)启动tensorboard服务:
在anaconda prompt窗口中指令:
tensorboard --logdir = '路径地址' --port 6006
(5)将地址输入浏览器打开即可,(点击刷新)
案例:
构建网络:
class Net(nn.Module):
def __init__(self):