pytorch删去预训练模型resnet18里最后一层模型

您可以使用以下代码删除ResNet18中的最后一层模型：

import torch.nn as nn
import torchvision.models as models

resnet18 = models.resnet18(pretrained=True)
modules = list(resnet18.children())[:-1]  # 删除最后一层
resnet18 = nn.Sequential(*modules)

这将创建一个新的ResNet18模型，其中包含除最后一层模型以外的所有层。注意，这个新的模型没有被预训练过，因此您需要在您的数据集上进行训练。