介绍
这篇文章给大家介绍怎么在Pytorch中修改ResNet模型全连接层,内容非常详细,感兴趣的小伙伴们可以参考借鉴,希望对大家能有所帮助。
最后一层的输出改一下,不需要加载前面层的权重,方法如下:
model =, torchvision.models.resnet18 (pretrained=False) num_fc_ftr =model.fc.in_features 时间=model.fc torch.nn.Linear (num_fc_ftr, 224) model =, nn.DataParallel(模型,device_ids=config.gpus)产品(设备)
首先模型结构是必须要传入的,然后把最后一层的输出改为自己所需的数目
关于怎么在Pytorch中修改ResNet模型全连接层就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看的到。