Function preTrainDArch return last model. May be is better to return best model + add stopAfterEpochs?