2021-09-26
构建神经网络步骤总结:
1、加载数据
使用的头文件:from torchvision import datasets, transforms
加载数据分为两部分:
加载训练集
加载测试集
二者差别不大,唯一却别就是训练集要将训练参数为true,测试集为false;
batchsz=32
cifar_train=datasets.CIFAR10('cifar',True,transform=transforms.Compose([
transforms.Resize((32,32)
原创
2021-09-26 21:18:11 ·
120 阅读 ·
0 评论