自己训练的pth预测全为0

[hgtlab]

作者您好，我用您的数据集运行了train，保存的pth文件用于测试中，但是test的时候预测输出的全为0，可是使用您训练好的pth测试时候，就会生成预测结果，请问一下是需要该哪部分代码吗

[hitachinsk]

我觉得这个可能是你配置的时候哪些地方有问题，本仓库的代码不需要改可以直接进行训练和测试。你check一下数据路径？

[NY1006CR]

我也是自己train的pth预测结果全为0，请问解决了吗

[Hang3Y]

俺也一样，test的结果是0，请问解决了吗？

[Saillxl]

我自己train的二分类预测结果全为0，请问解决了吗

[byhwhite]

--warmup --AdamW，训练时别忘了加上