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.")