jiangxiluning / FOTS.PyTorch

FOTS Pytorch Implementation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

yyr6661 opened this issue · comments

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

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