pytorch实现残差链接网络(resnet)

导入相关的库

import torch
import torch.nn as nn
import torch.optim as optim
import torchvision
import time
import os
import torch.nn.functional as F
import torch.backends.cudnn as cudnn
from torchvision import datasets, models, transforms
cudnn.benchmark = True
 
!pip install torchsummary

Requirement already satisfied: torchsummary in /opt/conda/lib/python3.7/site-packages (1.5.1)
WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv class="ansi-yellow-fg">

train_transform = transforms.Compose([
    transforms.RandomHorizontalFlip(),
    transforms.ToTensor()
])
 
val_transform = transforms.Compose([
    transforms.ToTensor()
])
 
batch_size = 64
 
trainset = torchvision.datasets.CIFAR10(root='./data', train=True,
                                        download=True, transform=train_transform)
trainloader = torch.utils.data.DataLoader(trainset, batch_size=batch_size,
                                          shuffle=True, num_workers=2)
 
testset = torchvision.datasets.CIFAR10(root='./data', train=False,
                                       download=True, transform=val_transform)
testloader = torch.utils.data.DataLoader(testset, batch_size=batch_size,
                                         shuffle=False, num_workers=2)

 

Files already downloaded and verified
Files already downloaded and verified