SaveTrainedModelHelper

fun SaveTrainedModelHelper(trainBatchSize: Int = 500, testBatchSize: Int = 1000)