resnet50预训练权重pytorch

你可以在PyTorch官方的模型库中找到ResNet-50的预训练权重。你可以使用以下代码来加载ResNet-50的预训练权重：

import torch
import torchvision.models as models

# 加载预训练权重
resnet = models.resnet50(pretrained=True)

# 保存权重
torch.save(resnet.state_dict(), 'resnet50_weights.pth')

在上面的代码中，我们使用`torchvision.models`模块中的`resnet50`函数来加载ResNet-50的预训练模型。然后，我们可以使用`state_dict()`方法来获取模型的权重，并使用`torch.save()`函数将权重保存到一个.pth文件中。

希望这能帮到你！如果你还有其他问题，请随时问。