search

chineseocr / table-detect

table detect(yolo) , table line(unet)
MIT License
236 stars 86 forks source link

test内存不释放，如何设置batch_size #11

Closed dcdethan closed 3 years ago

dcdethan commented 3 years ago

infer的时候在哪设置batch_size的大小呢？现在的问题是test图片多内存会增加不释放，我用的cpu进行的推理。