pytorch:cifar-10+resnet代码实现

import torch
from torch import nn
from torch.nn import functional as F
from torchvision import datasets
from torchvision import transforms
from torch.utils.data import DataLoader

trans = transforms.Compose((transforms.Resize(32),transforms.ToTensor()))
#要对从cifar中下载的图像进行一些转变，所以这里先初始化transform

cifar_train = datasets.CIFAR10('cifar',train = True,transform=trans)
#‘cifar’指我电脑里cifar数据集的保存路径，‘true’表示载入训练数据集，trans是我对载入图片进行的转变操作
cifar_train_batch = DataLoader(cifar_train,batch_size=30,shuffle=True)

#同上诉操作一样，只不过载入的是cifar的测试数据集
cifar_test = datasets.CIFAR10('cifar',train = False,transform=trans)
cifar_test_batch = DataLoader(cifar_test,batch_size=30,shuffle=True)

搭建resnet

首先搭建resblock

class resblock(nn.Module):
    def __init__(self,ch_in,ch_out,stride):
        super(resblock,self).__init__()
        self.conv_1 = nn.Conv2d(ch_in,ch_out,kernel_size=3,stride=stride,padding=1)
        self.bn_1 = nn.BatchNorm2d(ch_o