Browse Source

delete log infos

dev0.8.0
x54-729 2 years ago
parent
commit
bb52c8c3d2
1 changed files with 0 additions and 6 deletions
  1. +0
    -6
      fastNLP/core/drivers/torch_driver/ddp.py

+ 0
- 6
fastNLP/core/drivers/torch_driver/ddp.py View File

@@ -526,12 +526,6 @@ class TorchDDPDriver(TorchDriver):
raise RuntimeError(f"The `{fn}` attribute of model is not `Callable`.") raise RuntimeError(f"The `{fn}` attribute of model is not `Callable`.")
return fn, None return fn, None
elif fn in {"train_step", "evaluate_step"}: 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 return model, model.forward
else: else:
raise RuntimeError(f"There is no `{fn}` method in your model.") raise RuntimeError(f"There is no `{fn}` method in your model.")


Loading…
Cancel
Save