search

Layout-Parser / layout-model-training

The scripts for training Detectron2-based Layout Models on popular layout analysis datasets
202 stars 55 forks source link

alter for multi-gpu training #9

Closed TITC closed 2 years ago

TITC commented 2 years ago

I encounterd this error and found a solution at detectron2

@zsc1220 it seems the dataset is not registered when you use multiple GPUs. Where do you register the dataset? In train_net you might need to register it in the main() function.

lolipopshock commented 2 years ago

Thank you for your PR -- made some tiny tweaks to but others look great!