From 4e1a262712d3bef60d957cc6d890be589d963f6a Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Mon, 22 Nov 2021 14:07:25 -0800 Subject: [PATCH] remove debug prints --- deepspeed/runtime/config.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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: