pytorch mobilenetv3预训练模型下载
时间: 2023-07-27 07:06:53 浏览: 274
pytorch和ImageNet预训练模型中的MobileNetV3-Python开发
你可以使用以下代码下载PyTorch中的MobileNetV3预训练模型:
```python
import torch
model = torch.hub.load('pytorch/vision:v0.9.0', 'mobilenet_v3_large', pretrained=True)
```
这将下载并加载预训练的MobileNetV3 Large模型。如果你想要下载其他版本(如MobileNetV3 Small),只需将上述代码中的`mobilenet_v3_large`替换为`mobilenet_v3_small`即可。
阅读全文