fastNLP/fastNLP/core/utils/__init__.py

__all__ = [
    'cache_results',
    'is_jittor_module',
    'is_jittor_dataset',
    'jittor_collate_wraps',
    'paddle_to',
    'paddle_move_data_to_device',
    'get_paddle_device_id',
    'get_paddle_gpu_str',
    'is_in_paddle_dist',
    'is_in_fnlp_paddle_dist',
    'is_in_paddle_launch_dist',
    'is_paddle_module',
    'f_rich_progress',
    'torch_move_data_to_device',
    'is_torch_module',
    'get_oneflow_device',
    'oneflow_move_data_to_device',
    'is_oneflow_module',
    'is_in_oneflow_dist',
    'get_fn_arg_names',
    'auto_param_call',
    'check_user_specific_params',
    'dataclass_to_dict',
    'match_and_substitute_params',
    'apply_to_collection',
    'nullcontext',
    'pretty_table_printer',
    'Option',
    'deprecated',
    "flat_nest_dict",
    "f_tqdm_progress",

    "seq_len_to_mask"
]

from .cache_results import cache_results
from .jittor_utils import is_jittor_dataset, jittor_collate_wraps, is_jittor_module
from .paddle_utils import paddle_to, paddle_move_data_to_device, get_paddle_device_id, get_paddle_gpu_str, is_in_paddle_dist, \
    is_in_fnlp_paddle_dist, is_in_paddle_launch_dist, is_paddle_module
from .rich_progress import f_rich_progress
from .torch_utils import torch_move_data_to_device, is_torch_module
from .oneflow_utils import oneflow_move_data_to_device, is_oneflow_module, is_in_oneflow_dist, get_oneflow_device
from .utils import *
from .tqdm_progress import f_tqdm_progress
from .seq_len_to_mask import seq_len_to_mask