alter for multi-gpu training

fix: ValueError: With n_samples=0, test_size=None and train_size=0.8, the resulting train set will be empty. Adjust any of the aforementioned parameters.

Merge pull request #2 from nasheedyasin/master

Addressing the comments.

Update train_net.py

Update train_net.py

Update train_net.py

Merge branch 'master' of https://github.com/nasheedyasin/layout-model-training

Update train_net.py

Update cocosplit.py

Fixed issues with the coco-split utility

Update train_net.py

Add documentation for prima training

Add eval_and_save hook in trainer

Update Readme

Update gitignore

Add the script for prima training

Add the prima model configuration

Add training scripts

Rename folder and correct typo

Add library requirements

Add the prima conversion script

Add the cocosplit script

Initial Commit