pytorch加载部分预训练模型(解决问题的详细过程)
首先,给出代码地址https://github.com/jfzhang95/pytorch-deeplab-xception以及加载预训练模型的代码,做了些许修改
pretrain_dict = model_zoo.load_url('https://download.pytorch.org/models/resnet101-5d3b4d8f.pth')
model_dict = {}
state_dict = model.state_dict()
for k, v in p
原创
2020-06-12 16:18:43 ·
4436 阅读 ·
0 评论