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 = Dat