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

时间:2024-03-29 15:09:28

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

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

 

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

 

以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)

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

 

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

 

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