- 博客(2)
- 收藏
- 关注
原创 pytorch 网络搭建简要步骤
目录一、数据集加载二、搭建模型1.继承torch.nn.Module2.利用容器torch.nn.Sequential3.利用现有的预训练网络三、配置模型四、训练模型参考链接一、数据集加载用DataLoader和ImageFolder函数from torchvision.datasets import ImageFolderfrom torchvision import transformsfrom torch.utils.data import DataLoaderdata_transfor
2021-12-28 23:23:39
2339
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅
3