VikParuchuri / marker

Convert PDF to markdown quickly with high accuracy
https://www.datalab.to
GNU General Public License v3.0
14.65k stars 763 forks source link

How much GPU memory is required to train the layout segmenter model? #90

Closed codeants2012 closed 2 months ago

VikParuchuri commented 2 months ago

I trained it using DDP on 4x A6000s, but it could be trained with 24GB VRAM as well.