From bb52c8c3d28baf9a4fcbc2cff3c963b183dc88cb Mon Sep 17 00:00:00 2001 From: x54-729 <17307130121@fudan.edu.cn> Date: Tue, 13 Sep 2022 15:16:12 +0800 Subject: [PATCH] delete log infos --- fastNLP/core/drivers/torch_driver/ddp.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/fastNLP/core/drivers/torch_driver/ddp.py b/fastNLP/core/drivers/torch_driver/ddp.py index 923a544a..9330b9fc 100644 --- a/fastNLP/core/drivers/torch_driver/ddp.py +++ b/fastNLP/core/drivers/torch_driver/ddp.py @@ -526,12 +526,6 @@ class TorchDDPDriver(TorchDriver): raise RuntimeError(f"The `{fn}` attribute of model is not `Callable`.") return fn, None elif fn in {"train_step", "evaluate_step"}: - - logger.warning("\n\nfucking hei\n\n") - print(model) - print("\n\n") - print(type(model)) - print("\n\n") return model, model.forward else: raise RuntimeError(f"There is no `{fn}` method in your model.")