pytorch 加载已训练好的(.pth)格式模型
1 简介pytorch里有一些非常流行的网络如 resnet、wide_resnet101_2、squeezenet、densenet等,包括网络结构和训练好的模型。pytorch自带模型网址:https://pytorch-cn.readthedocs.io/zh/latest/torchvision/torchvision-models/按照官网加载预训练的模型:import torchvision.models as models # pretrained=True就可以使用预训练的模型
原创
2020-05-23 13:33:29 ·
28362 阅读 ·
0 评论