RuntimeError: CUDA out of memory. Tried to allocate 14.00 MiB (GPU 1; 10.76 GiB total capacity; 9.43 GiB already allocated; 7.44 MiB free; 9.75 GiB reserved in total by PyTorch)

JiaquanYe / TableMASTER-mmocr

2nd solution of ICDAR 2021 Competition on Scientific Literature Parsing, Task B.

Apache License 2.0

410 stars 100 forks source link

Open dream-incubation opened 11 months ago

dream-incubation commented 11 months ago

TableMaster在训练完成一轮后的推理时,会出现如上报错