diff --git a/deepspeed/runtime/config.py b/deepspeed/runtime/config.py index 9ac6f801d172..340a0a4a89e2 100755 --- a/deepspeed/runtime/config.py +++ b/deepspeed/runtime/config.py @@ -861,7 +861,7 @@ def __init__(self, config: Union[str, dict], mpu=None): def _initialize_params(self, param_dict): self.train_batch_size = get_train_batch_size(param_dict) - print(f"beginging get_train_batch_size = {get_train_batch_size}") + #print(f"beginning get_train_batch_size = {get_train_batch_size}") self.train_micro_batch_size_per_gpu = get_train_micro_batch_size_per_gpu( param_dict) self.gradient_accumulation_steps = get_gradient_accumulation_steps(param_dict) @@ -992,7 +992,7 @@ def _set_batch_related_parameters(self): micro_batch = self.train_micro_batch_size_per_gpu grad_acc = self.gradient_accumulation_steps - print(f"train_batch = {train_batch}, micro_batch={micro_batch}") + #print(f"train_batch = {train_batch}, micro_batch={micro_batch}") # all values are provided nothing needs to be set if train_batch is not None and micro_batch is not None and grad_acc is not None: