添加与训练模型，以densenet121为例

添加预训练模型的方法，自己记录

先把模型中的pretrained设置为True,如下

 

 

以densenet为例，在if下面添加

model_dict = model.state_dict()
# 将pretrained_dict里不属于model_dict的键剔除掉
pretrained_dict = {k: v for k, v in model_dict.items() if k in model_dict}
# 更新现有的model_dict
model_dict.update(pretrained_dict)
# 加载我们真正需要的state_dict
model.load_state_dict(model_dict)

 

注销：

_load_state_dict(model, model_urls[arch], progress)

 

这个也要设置为True,j就是你要调用的densenet121的方法，这个方法是我自己天机预训练模型之后改的名字，别介意