数据读取
Dataset 和 Dataloader
tensorboard
from torch.utils.tesnorboard import SummaryWriter
writer = SummaryWriter("logs")
# 两个常用的方法
writer.add_image() # 添加图像
writer.add_scalar() # 添加数据
writer.close()
transform
对图片进行变换
from torchvision import transforms
datasets + DataLoader
对图片进行变换
from torchvision import datasets
Module
最大池化层
保留数据特征但减少数据量
模型的保存与加载
```
方式1，模型结构+模型参数
torch.save(vgg16) torch.load()
方式2，模型参数
torch.save(vgg16.state_dict()) vgg16.load_state_dict()