pytorch VGG11识别cifar10数据集(训练+预测单张输入图片操作)

首先这是VGG的结构图，VGG11则是红色框里的结构，共分五个block，如红框中的VGG11第一个block就是一个conv3-64卷积层： 一，写VGG代码时，首先定义一个 vgg_block(n,in,out)方法，用来构建VGG中每个block中的卷积核和池化层： n是这个block中卷积层的数目，in是输入的通道数，out是输出的通道数 有了block以后，我们还需要一个方法把形成的block叠在一起，我们定义这个方法叫vgg_stack： def vgg_stack(num_convs, channels): # vgg_net = vgg_stack((1, 1, 2, 2