From eecef309cb12528cfa78d932a6f073afb43847e5 Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Tue, 20 Apr 2021 17:55:58 -0700 Subject: [PATCH] remove debug prints: (#986) --- deepspeed/runtime/zero/stage3.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deepspeed/runtime/zero/stage3.py b/deepspeed/runtime/zero/stage3.py index f8b52695..2b6e12ab 100755 --- a/deepspeed/runtime/zero/stage3.py +++ b/deepspeed/runtime/zero/stage3.py @@ -1418,13 +1418,13 @@ class FP16_DeepSpeedZeroOptimizer_Stage3(object): if torch.is_tensor(output): output = [output] else: - print(f'got UNKNOWN type {type(output)}') + #print(f'got UNKNOWN type {type(output)}') outputs = [] for name, val in vars(output).items(): if not name.startswith('__') and torch.is_tensor(val): outputs.append(val) output = outputs - print(f'convert output to {output}') + #print(f'convert output to {output}') for item in filter(lambda item: is_zero_param(item), output): if not any(id(item) in m._external_params for m in FWD_MODULE_STACK): -- GitLab