pytorch预训练模型下载
时间: 2023-11-12 17:57:25 浏览: 107
PyTorch 模型训练教程免费下载
可以通过以下代码下载PyTorch预训练模型:
```python
import torch
model = torch.hub.load('pytorch/vision', 'resnet50', pretrained=True)
```
其中,`resnet50`可以替换为其他模型名称,例如`resnet18`、`vgg16`等。
阅读全文
PyTorch 模型训练教程免费下载