本文介绍了两个常用的深度学习模型库：torchvision.models 和 pretrainedmodels，它们分别提供了如AlexNet、VGG、GoogleNet等经典模型及Xception、SENet等新模型的预训练版本，便于快速上手和应用。
1. torchvision的models
里面包含常用的深度模型：AlexNet、VGG、GoogleNet等
使用： model=models.resnet50()
2. pretrainedmodels
里面包含常用的深度模型：AlexNet、VGG、GoogleNet等，还包括一些新的模型Xception、SENet等
使用：args1=pretrainedmodels.__dict__['resnet50'](pretrained='imagenet').state_dict()
To be continue...
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
