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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章