autodl-projects/xautodl/config_utils/__init__.py

21 lines
764 B
Python
Raw Normal View History

2019-11-15 07:15:07 +01:00
##################################################
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2019 #
##################################################
2021-03-30 11:17:05 +02:00
# general config related functions
from .config_utils import load_config, dict2config, configure2str
2021-03-30 14:05:52 +02:00
2021-03-30 11:17:05 +02:00
# the args setting for different experiments
from .basic_args import obtain_basic_args
from .attention_args import obtain_attention_args
from .random_baseline import obtain_RandomSearch_args
from .cls_kd_args import obtain_cls_kd_args
from .cls_init_args import obtain_cls_init_args
2019-09-28 10:24:47 +02:00
from .search_single_args import obtain_search_single_args
2021-03-30 11:17:05 +02:00
from .search_args import obtain_search_args
2019-09-28 10:24:47 +02:00
# for network pruning
2021-03-30 11:17:05 +02:00
from .pruning_args import obtain_pruning_args
# utils for args
from .args_utils import arg_str2bool