after training bash train_prima.sh, model_final.pth is not getting generated into output folder.

Hi There,

I am trying to run the layout parser on custom data. It seems that after training on faster rcnn data I have observed that the mode_final.pth is not getting generated. 
Only below files are getting generated :
1. config.yaml
2. log.txt
3. metrics.json
4. events.out.tfevents.1699718647.aipths01.2797704.0
5. events.out.tfevents.1699720376.aiphts01.2799758.0

Please assist.
