how to choose number of epochs and batch size