add printing of training set size when running the training script