当前位置:   article > 正文

Pytorch_training2——network build+model (load+modify)

Pytorch_training2——network build+model (load+modify)

link

import torchvision
from torchvision import models
resnet50 = models.resnet50(pretrained=True) #pretrained=True 加载模型以及训练过的参数
print(resnet50)  # 打印输出观察一下resnet50到底是怎么样的结构
  • 1
  • 2
  • 3
  • 4

visit&modify nn_layer

resnet50=models.resnet50(pretrained=True)
num_in_ftrs=resnet.fc.in_features
  • 1
  • 2
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/很楠不爱3/article/detail/352977?site
推荐阅读
相关标签
  

闽ICP备14008679号