search

stefanklut / laypa

Layout analysis to find layout elements in documents (similar to P2PaLA)
MIT License
17 stars 4 forks source link

multithreading cpu training #18

Open rabeeqasem opened 7 months ago

rabeeqasem commented 7 months ago

how can we define in the YAML file the multi treathreading CPU and the number of workers to train ?

stefanklut commented 7 months ago

If you are training on CPU, I am not sure if multi threading is supported by detectron2. I am using their launch function for splitting the work over multiple GPUs. Perhaps you can find more in their documentation. To set the number of workers for the dataloader use:

DATALOADER:
  NUM_WORKERS: 16