# 导入模型
print("model loading")
path_model = "./static/model/resnet_.pt"
model=resnet()
model = model.load_state_dict(torch.load(path_model),False)
# model = model_loaded.load_state_dict(torch.load(path_model))
print("finish")
加载模型状态,首先要初始化模型状态,再加载状态
实际上,torch.load是加载state_dict而不是模型本身。
load_state_dict 的 strict 参数默认是 True :
表示预训练模型的层和自己定义的网络结构层严格对应相等(如层名和维度)
所以当我们修改了网络结构后,如果strict为True的时候就会报错
加载模型状态
https://blog.csdn.net/weixin_45727931/article/details/121516285)
strict参数
https://blog.csdn.net/weixin_45084253/article/details/123609392



