Spaces:
Build error
Build error
| # flake8: noqa | |
| # Copyright (c) OpenMMLab. All rights reserved. | |
| from .config import Config, ConfigDict, DictAction | |
| from .misc import (check_prerequisites, concat_list, deprecated_api_warning, | |
| has_method, import_modules_from_strings, is_list_of, | |
| is_method_overridden, is_seq_of, is_str, is_tuple_of, | |
| iter_cast, list_cast, requires_executable, requires_package, | |
| slice_list, to_1tuple, to_2tuple, to_3tuple, to_4tuple, | |
| to_ntuple, tuple_cast) | |
| from .path import (check_file_exist, fopen, is_filepath, mkdir_or_exist, | |
| scandir, symlink) | |
| from .progressbar import (ProgressBar, track_iter_progress, | |
| track_parallel_progress, track_progress) | |
| from .testing import (assert_attrs_equal, assert_dict_contains_subset, | |
| assert_dict_has_keys, assert_is_norm_layer, | |
| assert_keys_equal, assert_params_all_zeros, | |
| check_python_script) | |
| from .timer import Timer, TimerError, check_time | |
| from .version_utils import digit_version, get_git_hash | |
| try: | |
| import torch | |
| except ImportError: | |
| __all__ = [ | |
| 'Config', 'ConfigDict', 'DictAction', 'is_str', 'iter_cast', | |
| 'list_cast', 'tuple_cast', 'is_seq_of', 'is_list_of', 'is_tuple_of', | |
| 'slice_list', 'concat_list', 'check_prerequisites', 'requires_package', | |
| 'requires_executable', 'is_filepath', 'fopen', 'check_file_exist', | |
| 'mkdir_or_exist', 'symlink', 'scandir', 'ProgressBar', | |
| 'track_progress', 'track_iter_progress', 'track_parallel_progress', | |
| 'Timer', 'TimerError', 'check_time', 'deprecated_api_warning', | |
| 'digit_version', 'get_git_hash', 'import_modules_from_strings', | |
| 'assert_dict_contains_subset', 'assert_attrs_equal', | |
| 'assert_dict_has_keys', 'assert_keys_equal', 'check_python_script', | |
| 'to_1tuple', 'to_2tuple', 'to_3tuple', 'to_4tuple', 'to_ntuple', | |
| 'is_method_overridden', 'has_method' | |
| ] | |
| else: | |
| from .env import collect_env | |
| from .logging import get_logger, print_log | |
| from .parrots_jit import jit, skip_no_elena | |
| from .parrots_wrapper import ( | |
| TORCH_VERSION, BuildExtension, CppExtension, CUDAExtension, DataLoader, | |
| PoolDataLoader, SyncBatchNorm, _AdaptiveAvgPoolNd, _AdaptiveMaxPoolNd, | |
| _AvgPoolNd, _BatchNorm, _ConvNd, _ConvTransposeMixin, _InstanceNorm, | |
| _MaxPoolNd, get_build_config, is_rocm_pytorch, _get_cuda_home) | |
| from .registry import Registry, build_from_cfg | |
| from .trace import is_jit_tracing | |
| __all__ = [ | |
| 'Config', 'ConfigDict', 'DictAction', 'collect_env', 'get_logger', | |
| 'print_log', 'is_str', 'iter_cast', 'list_cast', 'tuple_cast', | |
| 'is_seq_of', 'is_list_of', 'is_tuple_of', 'slice_list', 'concat_list', | |
| 'check_prerequisites', 'requires_package', 'requires_executable', | |
| 'is_filepath', 'fopen', 'check_file_exist', 'mkdir_or_exist', | |
| 'symlink', 'scandir', 'ProgressBar', 'track_progress', | |
| 'track_iter_progress', 'track_parallel_progress', 'Registry', | |
| 'build_from_cfg', 'Timer', 'TimerError', 'check_time', 'SyncBatchNorm', | |
| '_AdaptiveAvgPoolNd', '_AdaptiveMaxPoolNd', '_AvgPoolNd', '_BatchNorm', | |
| '_ConvNd', '_ConvTransposeMixin', '_InstanceNorm', '_MaxPoolNd', | |
| 'get_build_config', 'BuildExtension', 'CppExtension', 'CUDAExtension', | |
| 'DataLoader', 'PoolDataLoader', 'TORCH_VERSION', | |
| 'deprecated_api_warning', 'digit_version', 'get_git_hash', | |
| 'import_modules_from_strings', 'jit', 'skip_no_elena', | |
| 'assert_dict_contains_subset', 'assert_attrs_equal', | |
| 'assert_dict_has_keys', 'assert_keys_equal', 'assert_is_norm_layer', | |
| 'assert_params_all_zeros', 'check_python_script', | |
| 'is_method_overridden', 'is_jit_tracing', 'is_rocm_pytorch', | |
| '_get_cuda_home', 'has_method' | |
| ] | |