Layout-Parser / layout-model-training

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

Loss becoming infinite #11

Open Mohdwajtech opened 2 years ago

Mohdwajtech commented 2 years ago

Hi, I have tried training on new set of 80 images following the steps but somewhere in between the training I get this error which I have just attached within this issue as an image. I tried it multiple times but looks like it gets stuck at any random iteration. The only solution is if I keep the dataset small i.e. 10 - 15 pages I am sure not to get this error. Is there a possible workaround to this issue training_error ?

RGuven commented 1 year ago

did you solve the problem?