【Pytorch学习】TensorBoard 可视化模型，以及训练过程

 【Pytorch学习】TensorBoard 可视化模型，以及训练过程
1，TensorBoard 设置与写入
现在，我们将设置 TensorBoard，从torch.utils导入tensorboard并定义SummaryWriter，这是将信息写入 TensorBoard 的关键对象。
#首先定义模型
class Net(nn.Module):
def __init__(self):
super(Net, self).__init__()
self.conv1 = nn.Conv2d(1, 6, 5)
self.pool = nn.MaxPool2d(2, 2)
self.conv2 = nn.Conv2d(6, 16, 5)
self.fc1 = nn.Linear(16 * 4 * 4, 120)
self.fc2 = nn.Linear(120, 84)
self.fc3 = nn.Linear(84, 10

【Pytorch学习】TensorBoard 可视化模型，以及训练过程最先出现在Python成神之路。