当前位置: 代码迷 >> 综合 >> load_state_dict ‘NoneType‘ object has no attribute ‘copy‘
  详细解决方案

load_state_dict ‘NoneType‘ object has no attribute ‘copy‘

热度:90   发布时间:2024-01-12 18:20:56.0

训练完之后,加载最优模型这,torch.load(path) 加载的模型一直是空的

1. 最先感觉路径写错,改了多次路径,发现不是路径问题

2. 看了看最优模型大小只有140多Bytes,和以前300多MB相隔甚远

    自己打印了下保存的最优模型还是None,说明训练完保存的最优模型是空的

3. 定位到训练模块,保存最优模型这里有问题,发现保存条件弄错了一个细节

    目标是某轮f1值大于最优f1值时保存下来此模型,结果条件写成小于了,

    由于初始最优f1值是0,而每轮的验证集f1值永远大于0,所以永远不会保存什么模型.....

 

pretrained_model_dict = torch.load( path )

print( type( pretrained_model_dict ) )

# class: NoneType  

  相关解决方案