diff --git a/fastNLP/core/__init__.py b/fastNLP/core/__init__.py index 28479b05..10472917 100644 --- a/fastNLP/core/__init__.py +++ b/fastNLP/core/__init__.py @@ -69,6 +69,12 @@ __all__ = [ "JittorMPIDriver", "OneflowSingleDriver", "OneflowDDPDriver", + "torch_seed_everything", + "paddle_seed_everything", + "oneflow_seed_everything", + "torch_move_data_to_device", + 'paddle_move_data_to_device', + 'oneflow_move_data_to_device', # log "logger",