From ec3f27f8a2af691c466ea448105974bec3b7e4c6 Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Tue, 20 Apr 2021 17:05:33 -0700 Subject: [PATCH] remove debug prints: --- 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 f8b526952de8..2b6e12abd84b 100755 --- a/deepspeed/runtime/zero/stage3.py +++ b/deepspeed/runtime/zero/stage3.py @@ -1418,13 +1418,13 @@ def _post_forward_module_hook(module, input, output): 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):