Created by: backyes

enable num_batches_regularization for parameters config in trainer_config_helpers API.

// TODO: add more annotations for num_batches_regularization.