关于读取模型权重的问题 about iim

[gjy3035]

strict=True
才能保证参数被完全加载。
如果改为True报错，可以看看这个问题
#5

[Veefas]

strict=True
才能保证参数被完全加载。
如果改为True报错，可以看看这个问题
#5

我是用4核GPU训练的，test的时候试了4核和1核，strict改为True以后还是不行，预测的像素点全是0。。按理说权重加载正确，效果应该和验证集差不多，还是有些费解

[taohan10200]

您好，我们刚才已经在SHHB数据集上重新验证了，对于训练过程中保存的模型是可以直接使用test.py进行预测的，请问您是不是clone的最新版本，以及是不是在后续改动过程中引起了新的问题？
训练过程中保存的模型：

在test.py中添加要测试的模型路径，以及修改数据集名称等相关参数：

在测试集上的预测结果：./saved_exp_results/SHHB_HR_Net_test.txt'

[Veefas]

您好，我早上用SHHB数据集上训练验证，的确没有问题。但之前那一批训练的结果就是预测像素点为0，十分的迷幻，应该是我改动了哪个地方导致的，暂时不纠结这个点了，anyway，感谢作者大大！