fastNLP/fastNLP/core/utils/__init__.py

__all__ = [
    'cache_results',
    'is_jittor_dataset',
    'jittor_collate_wraps',
    'get_device_from_visible',
    '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',
    'f_rich_progress',
    'torch_paddle_move_data_to_device',
    'torch_move_data_to_device',
    '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',
    'seq_len_to_mask',
    "flat_nest_dict"
]

from .cache_results import cache_results
from .jittor_utils import is_jittor_dataset, jittor_collate_wraps
from .paddle_utils import get_device_from_visible, 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
from .rich_progress import f_rich_progress
from .torch_paddle_utils import torch_paddle_move_data_to_device
from .torch_utils import torch_move_data_to_device
from .utils import *