文章目录
- 模型训练的模板
- 综合案例-Pytorch 官网demo
- 优化
- 记录日志
- 解析日志
- 增加tensorboard数据记录
- 保存训练曲线
- 模型参数可视化
- 增加wandb数据记录
模型训练的模板
综合案例-Pytorch 官网demo
pytorch 官网tutorial-quickstart
https://blog.csdn.net/weixin_39107270/article/details/124382038
import torch
from torch import nn
from torch.utils.data import DataLoader
from torchvision import datasets
from torchvision.transforms