Resnet18 PyTorch复现代码

以下是一个简单的Resnet18的PyTorch复现代码，包括了网络结构的定义和训练部分的代码。代码中使用了CIFAR10数据集作为示例数据集，可以根据自己的需求更改数据集。

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

# 定义ResNet基本块类
class BasicBlock(nn.Module):
    def __init__(self, in_channels, out_channels, stride):
        super(BasicBlock, self).__init__()
        self.conv1 = nn.Conv2d(in_channels, out_channels, kernel_size=3, stride=stride, padding=1, bias=False)
        self.bn1 = nn.BatchNorm2