Move training to gpu and add early stopping

Merge request reports

Loading