文章目录
1、tensorboard使用
from torch.utils.tensorboard import SummaryWriter
log_dir = "路径"
writer = SummaryWriter(log_dir = log_dir ,filename_suffix = "_kernel")
会生成events文件
生成后，在终端打开到events文件对应的路径，执行
tensorboard --logdir=./
就会生成相应的链接
2、构建模型
0、一些参数
1、首先是数据
预处理transforms
读取数据
数据加载器
2、然后是模型
模型的预训练，微调，传入GPU
3、损失函数
4、优化器
优化器选择
学习率使用
5、训练
读取数据
传入GPU
前向传播-训练模型
反向传播-优化器置零（把模型中参数的梯度设为0）、损失的计算
优化器更新
统计分类情况
打印损失和精度信息
验证集进行验证