pytorch的一些库

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...

发布了31 篇原创文章 · 获赞 8 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weijie_home/article/details/104842287
今日推荐