from torchvision.models.utils import load_state_dict_from_url需要什么库
时间: 2023-11-28 10:04:04 浏览: 82
nfs-utils-1.1.1.tar.gz_nfs utils_nfs-utils
需要安装PyTorch库。`load_state_dict_from_url` 是 PyTorch 提供的一个函数,用于从指定的URL加载预训练模型的权重。这个函数可以在torchvision.models中使用,可以通过pip安装。可以使用以下命令安装PyTorch:
```
pip install torch torchvision
```
阅读全文