摘要:
很久没在CSDN写过东西了, 今天被一个bug气到自己,太愚蠢了,记录一下。 错误代码如下: # Error code! model = model.load_state_dict(state_dict_var) out = model(input) 网上有一些问题是因为 存的model, 却用st 阅读全文
摘要:
出现该报错的原因是:训练保存的模型里包含了网络结构,结构的model类直接包含了路径地址model.yolo.Model 解决办法就是重新在官方项目代码中加载模型,并再次只保存其中的模型权重,注意不要保存模型结构。 保存方法如下: trans = torch.load('你的模型路径') troch 阅读全文