facebookresearch / nougat

Implementation of Nougat Neural Optical Understanding for Academic Documents
https://facebookresearch.github.io/nougat/
MIT License
8.81k stars 560 forks source link

[Feature Request] Set number of CPU workers in the CLI #149

Open MereHappyJaZZ opened 11 months ago

MereHappyJaZZ commented 11 months ago

Hi!

On Windows, python does not support multiprocessing workers above above 61, see e.g. https://stackoverflow.com/questions/65252807/multiprocessing-pool-pool-on-windows-cpu-limit-of-63

Can we get a flag to the nougat CLI that sets a "num_workers" to help with this issue?

Thanks!