模型在训练时损失很大

dk-liang / FIDTM

[IEEE TMM] Focal Inverse Distance Transform Maps for Crowd Localization

MIT License

173 stars 42 forks source link

Closed waric01 closed 1 year ago

waric01 commented 2 years ago

老师，您好！我在使用你的模型训练时，发现损失很大，第一轮loss高达几十万，随着训练轮次增加，loss依然好几千，请问一下是什么原因呢，谢谢！

dk-liang commented 2 years ago

你好，这是正常现象，因为我算的loss是pixel层面的MSE loss，用的是sum，不是mean

waric01 commented 2 years ago

好的，非常感谢老师您在百忙中抽时间解答，十分感激和感谢！