tensorboard的学习网址:Visualizing Models, Data, and Training with TensorBoard — PyTorch Tutorials 1.12.0+cu102 documentation
在上面的红框即可找到对应的链接网址
tensorboard主要使用的功能
一. 保存网络的结构图(在tensorboard的GRAPHS中)
1.1. 错误的操作步骤
真奇怪,下面的一系列操作竟然不能在tensorboard中显示模型结构,无语了。
import torch
import numpy as np
from torch.utils.tensorboard import SummaryWriter
import torch.nn as nn
class mymodel(nn.Module):
def __init__(self):
super(mymodel, self).__init__()
self.model1 = nn.Sequential(
nn.Conv2d(1, 3, kernel_size=(3,3), stride=(1,1),