tensorboard可视化_CoLab的PyTorch环境下使用TensorBoard实现可视化

首先你有定义一个网络：

# 以上省略网络的定义
net = Net()

使用SummaryWriter和tensorboard，这种方式适用于CoLab和一般开发

• SummaryWriter类是使用log日志进行TensorBoard可视化的主要入口^[1]

from torch.utils.tensorboard import SummaryWriter

# default `log_dir` is "runs" - we'll be more specific here
writer = SummaryWriter('runs/fashion_mnist_experiment_1')

• 加载TensorBoard在CoLab环境中的插件

# Load the TensorBoard notebook extension
%load_ext tensorboard

• 打开TensorBoard面板，此时还没有读入log日志，需