添加预训练模型的方法,自己记录
先把模型中的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的方法,这个方法是我自己天机预训练模型之后改的名字,别介意