EAST icon indicating copy to clipboard operation
EAST copied to clipboard
verified publisher icon SakuraRiven

训练结果差

Open Shualite opened this issue 5 years ago • 1 comments

ICDAR2015数据集，训了50个epoch以后，eval结果任然是0.05左右。 而且分类loss和刚开始训的时候一样一直是0.99左右（如果我没记错的话）。 想请问一下，问题出在dataloader上还是损失函数那个部分？ 感谢！

Shualite avatar Jul 18 '20 07:07 Shualite

后来发现Dice loss可能开始不容易收敛……可以使用CE loss作为辅助……

SakuraRiven avatar Aug 18 '20 08:08 SakuraRiven