ResNet18模型详解+云端实战：理论实践结合，1元体验

1. 为什么选择ResNet18作为入门模型

当你刚开始学习深度学习时，可能会被各种复杂的网络结构搞得晕头转向。ResNet18就像是一个"恰到好处"的入门选择——它足够简单让你理解基本原理，又足够强大能完成实际任务。

想象一下盖房子。传统神经网络就像不断往上垒砖块，层数越多房子越高（性能越好），但超过一定高度后，房子反而容易倒塌（梯度消失/爆炸）。ResNet的创新在于加入了"电梯"（残差连接），让信息可以跨层直达，解决了深层网络难以训练的问题。

ResNet18特别适合初学者因为：

结构清晰：仅18层，每层作用明确
资源友好：相比ResNet50/101，对计算资源要求低
效果不错：在CIFAR-10上轻松达到80%+准确率
应用广泛：是许多实际项目的基准模型

2. ResNet18结构拆解：从输入到输出

让我们把ResNet18拆开来看，就像拆解一台精密的相机：

2.1 整体架构

ResNet18可以分成5个部分：

入口层：像相机的镜头，负责初步处理输入图像
4个残差块组：像相机的不同功能模块（对焦、测光等），每组完成特定特征提取
全连接层：像相机的图像处理器，最终输出分类结果

2.2 核心创新：残差块

残差块是ResNet的灵魂。想象你在学习骑自行车：

传统网络：每次都要从头学习平衡
ResNet：每次只需学习与当前状态的"差异"（残差）

用代码表示一个基本残差块：

import torch.nn as nn class BasicBlock(nn.Module): def __init__(self, in_channels, out_channels, stride=1): super().__init__() self.conv1 = nn.Conv2d(in_channels, out_channels, kernel_size=3, stride=stride, padding=1) self.bn1 = nn.BatchNorm2d(out_channels) self.relu = nn.ReLU() self.conv2 = nn.Conv2d(out_channels, out_channels, kernel_size=3, padding=1) self.bn2 = nn.BatchNorm2d(out_channels) # 当输入输出尺寸不一致时需要调整 self.shortcut = nn.Sequential() if stride != 1 or in_channels != out_channels: self.shortcut = nn.Sequential( nn.Conv2d(in_channels, out_channels, kernel_size=1, stride=stride), nn.BatchNorm2d(out_channels) ) def forward(self, x): residual = self.shortcut(x) out = self.conv1(x) out = self.bn1(out) out = self.relu(out) out = self.conv2(out) out = self.bn2(out) out += residual # 关键残差连接 out = self.relu(out) return out

3. 云端实战：1元体验ResNet18训练

现在我们来实际训练一个ResNet18模型，识别CIFAR-10数据集中的物体（飞机、汽车、鸟等10类）。

3.1 环境准备

在CSDN算力平台，选择预置的PyTorch镜像（已包含CUDA加速），按小时计费，新用户1元即可体验：

登录CSDN算力平台
选择"PyTorch 1.12 + CUDA 11.3"镜像
配置GPU资源（T4显卡足够）
启动实例

3.2 完整训练代码

创建resnet18_cifar10.py文件，复制以下代码：

import torch import torch.nn as nn import torch.optim as optim import torchvision import torchvision.transforms as transforms from torchvision.models import resnet18 from tqdm import tqdm # 1. 数据准备 transform = transforms.Compose([ transforms.RandomHorizontalFlip(), transforms.RandomCrop(32, padding=4), transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) ]) trainset = torchvision.datasets.CIFAR10(root='./data', train=True, download=True, transform=transform) trainloader = torch.utils.data.DataLoader(trainset, batch_size=128, shuffle=True, num_workers=2) testset = torchvision.datasets.CIFAR10(root='./data', train=False, download=True, transform=transform) testloader = torch.utils.data.DataLoader(testset, batch_size=100, shuffle=False, num_workers=2) # 2. 模型调整（原始ResNet18是为ImageNet设计的） model = resnet18(pretrained=False) model.conv1 = nn.Conv2d(3, 64, kernel_size=3, stride=1, padding=1, bias=False) # CIFAR-10图片较小 model.fc = nn.Linear(512, 10) # 输出10类 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") model = model.to(device) # 3. 训练配置 criterion = nn.CrossEntropyLoss() optimizer = optim.SGD(model.parameters(), lr=0.1, momentum=0.9, weight_decay=5e-4) scheduler = optim.lr_scheduler.MultiStepLR(optimizer, milestones=[30, 60], gamma=0.1) # 4. 训练循环 for epoch in range(80): model.train() running_loss = 0.0 for inputs, labels in tqdm(trainloader, desc=f'Epoch {epoch+1}'): inputs, labels = inputs.to(device), labels.to(device) optimizer.zero_grad() outputs = model(inputs) loss = criterion(outputs, labels) loss.backward() optimizer.step() running_loss += loss.item() scheduler.step() # 每个epoch测试准确率 model.eval() correct = 0 total = 0 with torch.no_grad(): for inputs, labels in testloader: inputs, labels = inputs.to(device), labels.to(device) outputs = model(inputs) _, predicted = torch.max(outputs.data, 1) total += labels.size(0) correct += (predicted == labels).sum().item() print(f'Epoch {epoch+1}, Loss: {running_loss/len(trainloader):.3f}, Acc: {100*correct/total:.2f}%') # 5. 保存模型 torch.save(model.state_dict(), 'resnet18_cifar10.pth')

3.3 运行与监控

在终端执行：

python resnet18_cifar10.py

你会看到类似这样的输出：

Epoch 1: 100%|██████████| 391/391 [00:15<00:00, 25.12it/s] Epoch 1, Loss: 1.763, Acc: 38.52% Epoch 2: 100%|██████████| 391/391 [00:14<00:00, 26.34it/s] Epoch 2, Loss: 1.215, Acc: 56.89% ... Epoch 80: 100%|██████████| 391/391 [00:14<00:00, 26.89it/s] Epoch 80, Loss: 0.112, Acc: 85.67%