在PyTorch中，加载预训练权重非常简单。首先，确保你已经安装了PyTorch库。然后，按照以下步骤操作：
import torch
import torchvision.models as models
model = models.resnet18(pretrained=True)
如果你想使用其他预训练模型，只需将models.resnet18()替换为所需模型的函数，例如models.vgg16(pretrained=True)。
.pth或.pt格式），你可以使用以下代码加载它：weights = torch.load('path/to/your/weights.pth')
model.load_state_dict(weights)
现在，你已经成功加载了预训练权重到PyTorch模型中。你可以继续训练模型或在测试集上评估模型性能。
