训练完之后,加载最优模型这,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