首先,我们需要导入必要的库:import torch import torchvision from torchvision import transforms from torch.utils.data import DataLoader from torchvision.models import ReXNet from torch import nn, optim 接下来,我们需要定义数据预处理:transform = transforms.Compose([transforms.Resize(256), transforms.CenterCrop(224), transforms.T