请问为何训练过程中显存占用持续增加?
yyr6661 opened this issue · comments
yyr6661 commented
ZachZhang commented
+1
icdar2015数据集训练中我也遇到了,可能哪里显存溢出了?而且识别的loss一直无法下降
bianlongpeng commented
you can turn down your batch size,may be 32 to 16,for me it soved this problem!