Pytorch框架使用预训练的ResNet18模型如何下载其权重

你可以使用 PyTorch 的 `torchvision` 库来下载预训练的 ResNet18 模型。以下是一些代码示例，可以帮助你完成下载：

import torch
import torchvision

# 下载 ResNet18 模型的预训练权重
model = torchvision.models.resnet18(pretrained=True)

# 保存模型权重
torch.save(model.state_dict(), 'resnet18_weights.pth')

在上面的代码中，我们首先导入 `torch` 和 `torchvision` 模块。接下来，我们使用 `torchvision.models.resnet18()` 函数来下载预训练的 ResNet18 模型。该函数具有一个可选参数 `pretrained`，默认值为 `False`，将其设置为 `True` 以下载预训练权重。

最后，我们使用 `torch.save()` 将模型的权重保存到一个文件中。在这里，我们将其保存为名为 `resnet18_weights.pth` 的文件。