def METHOD_NAME(self, batch_size=None):
    """
    Update parameters like batch size, num_workers of the dataloader
    From the parallelism config file.
    """
    if batch_size:
        self.batch_sampler.batch_size = batch_size
    else:
        self.load_config(self.config_file)