上传者: 38580759
|
上传时间: 2021-12-23 22:25:54
|
文件大小: 104KB
|
文件类型: -
有一些非常流行的网络如 resnet、squeezenet、densenet等在pytorch里面都有,包括网络结构和训练好的模型。
pytorch自带模型网址:https://pytorch-cn.readthedocs.io/zh/latest/torchvision/torchvision-models/
按官网加载预训练好的模型:
import torchvision.models as models
# pretrained=True就可以使用预训练的模型
resnet18 = models.resnet18(pretrained=True)
print(resnet18)
报错如