From 3ea094298000c6148dfd5bcc52d65e88e85fe582 Mon Sep 17 00:00:00 2001 From: yhcc Date: Wed, 6 Jul 2022 22:34:27 +0800 Subject: [PATCH] export more function --- fastNLP/core/__init__.py | 6 ++++++ 1 file changed, 6 insertions(+) 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",