search

LBH1024 / CAN

When Counting Meets HMER: Counting-Aware Network for Handwritten Mathematical Expression Recognition (ECCV’2022 Poster).
MIT License
364 stars 59 forks source link

内存溢出，在使用本代码在学校服务器上运行时，出现内存溢出的清况，大家有遇到这种情况嘛，该如何解决呢？ #30

Open zengyangzang opened 1 year ago

Vegemo-bear commented 1 year ago

有的，这个需要batchsize设置小一些，如果最开始跑占用的显存是5000MB左右，最后稳定的时候是8200多MB。

SuperHHzy commented 1 year ago

降低batchsize，用apex半精度训练。

kingqiuol commented 1 year ago

多卡的话，就DDP，半精度在数据量足够多用(10w以上吧)，不然会不收敛