FOTS.PyTorch icon indicating copy to clipboard operation
FOTS.PyTorch copied to clipboard

请问为何训练过程中显存占用持续增加?

Open yyr6661 opened this issue 5 years ago • 2 comments

yyr6661 avatar Apr 14 '20 07:04 yyr6661

+1 icdar2015数据集训练中我也遇到了,可能哪里显存溢出了?而且识别的loss一直无法下降

KaiiZhang avatar Jun 17 '20 12:06 KaiiZhang

you can turn down your batch size,may be 32 to 16,for me it soved this problem!

bianlongpeng avatar May 25 '21 12:05 bianlongpeng