torch内置模型

torch内置模型

使用

打印支持的模型:

print(dir(torchvision.models))

下载模型,加上pretrained=True参数,会自动下载预训练好的模型:

resnet18 = models.resnet18(pretrained=True)

Downloading: "https://download.pytorch.org/models/resnet18-f37072fd.pth" to /home/guru_ge/.cache/torch/hub/checkpoints/resnet18-f37072fd.pth

打印网络层:

print(resnet18)

resnet

resnet
resnet

性能

网络 flops(G) params(M)
 vgg16
15.484
138.358
 resnet50  
4.112
25.557
resnet18
1.819
11.690
mobilnet_v1 0.57 4.25
mobilenet_v2
0.314
3.505
mobilenet_v3_small
0.059
2.543
mobilenet_v3_large
0.225
5.483

posted @ 2022-01-07 11:53  bairuiworld  阅读(212)  评论(0编辑  收藏  举报