From 94a149b33f559ebd4cc1fdbb5f702b47079e2705 Mon Sep 17 00:00:00 2001 From: D-X-Y <280835372@qq.com> Date: Tue, 18 May 2021 14:08:00 +0000 Subject: [PATCH] Move to xautodl --- .gitmodules | 3 + .latent-data/NAS-Bench-201 | 1 + CHANGE-LOG.md | 5 +- docs/README_CN.md | 2 +- docs/requirements.txt | 8 +-- exps/LFNA/lfna.py | 30 +++++----- lib | 1 + setup.py | 59 +++++++++++++++++++ xautodl/__init__.py | 6 ++ {lib => xautodl}/config_utils/__init__.py | 0 {lib => xautodl}/config_utils/args_utils.py | 0 .../config_utils/attention_args.py | 0 {lib => xautodl}/config_utils/basic_args.py | 0 .../config_utils/cls_init_args.py | 0 {lib => xautodl}/config_utils/cls_kd_args.py | 0 {lib => xautodl}/config_utils/config_utils.py | 0 {lib => xautodl}/config_utils/pruning_args.py | 0 .../config_utils/random_baseline.py | 0 {lib => xautodl}/config_utils/search_args.py | 0 .../config_utils/search_single_args.py | 0 {lib => xautodl}/config_utils/share_args.py | 0 .../datasets/DownsampledImageNet.py | 0 {lib => xautodl}/datasets/LandmarkDataset.py | 0 .../datasets/SearchDatasetWrap.py | 0 {lib => xautodl}/datasets/__init__.py | 0 .../datasets/get_dataset_with_transform.py | 0 .../datasets/landmark_utils/__init__.py | 0 .../datasets/landmark_utils/point_meta.py | 0 {lib => xautodl}/datasets/math_adv_funcs.py | 0 {lib => xautodl}/datasets/math_base_funcs.py | 0 {lib => xautodl}/datasets/math_core.py | 0 .../datasets/math_dynamic_funcs.py | 0 {lib => xautodl}/datasets/synthetic_core.py | 0 {lib => xautodl}/datasets/synthetic_env.py | 0 .../datasets/synthetic_example.py | 0 {lib => xautodl}/datasets/synthetic_utils.py | 0 {lib => xautodl}/datasets/test_utils.py | 0 {lib => xautodl}/log_utils/__init__.py | 0 {lib => xautodl}/log_utils/logger.py | 0 {lib => xautodl}/log_utils/meter.py | 0 {lib => xautodl}/log_utils/pickle_wrap.py | 0 {lib => xautodl}/log_utils/time_utils.py | 0 {lib => xautodl}/models/CifarDenseNet.py | 0 {lib => xautodl}/models/CifarResNet.py | 0 {lib => xautodl}/models/CifarWideResNet.py | 0 .../models/ImageNet_MobileNetV2.py | 0 {lib => xautodl}/models/ImageNet_ResNet.py | 0 {lib => xautodl}/models/SharedUtils.py | 0 {lib => xautodl}/models/__init__.py | 0 .../models/cell_infers/__init__.py | 0 {lib => xautodl}/models/cell_infers/cells.py | 0 .../models/cell_infers/nasnet_cifar.py | 0 .../models/cell_infers/tiny_network.py | 0 {lib => xautodl}/models/cell_operations.py | 0 .../models/cell_searchs/__init__.py | 0 .../models/cell_searchs/_test_module.py | 0 .../models/cell_searchs/generic_model.py | 0 .../models/cell_searchs/genotypes.py | 0 .../models/cell_searchs/search_cells.py | 0 .../models/cell_searchs/search_model_darts.py | 0 .../cell_searchs/search_model_darts_nasnet.py | 0 .../models/cell_searchs/search_model_enas.py | 0 .../cell_searchs/search_model_enas_utils.py | 0 .../models/cell_searchs/search_model_gdas.py | 0 .../search_model_gdas_frc_nasnet.py | 0 .../cell_searchs/search_model_gdas_nasnet.py | 0 .../cell_searchs/search_model_random.py | 0 .../models/cell_searchs/search_model_setn.py | 0 .../cell_searchs/search_model_setn_nasnet.py | 0 {lib => xautodl}/models/clone_weights.py | 0 {lib => xautodl}/models/initialization.py | 0 .../models/shape_infers/InferCifarResNet.py | 0 .../shape_infers/InferCifarResNet_depth.py | 0 .../shape_infers/InferCifarResNet_width.py | 0 .../shape_infers/InferImagenetResNet.py | 0 .../models/shape_infers/InferMobileNetV2.py | 0 .../models/shape_infers/InferTinyCellNet.py | 0 .../models/shape_infers/__init__.py | 0 .../models/shape_infers/shared_utils.py | 0 .../models/shape_searchs/SearchCifarResNet.py | 0 .../shape_searchs/SearchCifarResNet_depth.py | 0 .../shape_searchs/SearchCifarResNet_width.py | 0 .../shape_searchs/SearchImagenetResNet.py | 0 .../shape_searchs/SearchSimResNet_width.py | 0 .../models/shape_searchs/SoftSelect.py | 0 .../models/shape_searchs/__init__.py | 0 .../generic_size_tiny_cell_model.py | 0 {lib => xautodl}/models/shape_searchs/test.py | 0 {lib => xautodl}/models/xcore.py | 0 {lib => xautodl}/nas_201_api/__init__.py | 0 {lib => xautodl}/nas_201_api/api_201.py | 0 {lib => xautodl}/nas_201_api/api_utils.py | 0 .../nas_infer_model/DXYs/CifarNet.py | 0 .../nas_infer_model/DXYs/ImageNet.py | 0 .../nas_infer_model/DXYs/__init__.py | 0 .../nas_infer_model/DXYs/base_cells.py | 0 .../nas_infer_model/DXYs/construct_utils.py | 0 .../nas_infer_model/DXYs/genotypes.py | 0 .../nas_infer_model/DXYs/head_utils.py | 0 {lib => xautodl}/nas_infer_model/__init__.py | 0 .../nas_infer_model/operations.py | 0 {lib => xautodl}/procedures/__init__.py | 0 {lib => xautodl}/procedures/advanced_main.py | 0 {lib => xautodl}/procedures/basic_main.py | 0 {lib => xautodl}/procedures/eval_funcs.py | 0 {lib => xautodl}/procedures/funcs_nasbench.py | 0 {lib => xautodl}/procedures/metric_utils.py | 0 {lib => xautodl}/procedures/optimizers.py | 0 {lib => xautodl}/procedures/q_exps.py | 0 {lib => xautodl}/procedures/search_main.py | 0 {lib => xautodl}/procedures/search_main_v2.py | 0 {lib => xautodl}/procedures/simple_KD_main.py | 0 {lib => xautodl}/procedures/starts.py | 0 {lib => xautodl}/spaces/__init__.py | 0 {lib => xautodl}/spaces/basic_op.py | 0 {lib => xautodl}/spaces/basic_space.py | 0 {lib => xautodl}/trade_models/__init__.py | 0 .../trade_models/naive_v1_model.py | 0 .../trade_models/naive_v2_model.py | 0 .../trade_models/quant_transformer.py | 0 {lib => xautodl}/trade_models/transformers.py | 0 {lib => xautodl}/utils/__init__.py | 0 {lib => xautodl}/utils/affine_utils.py | 0 {lib => xautodl}/utils/evaluation_utils.py | 0 {lib => xautodl}/utils/flop_benchmark.py | 0 {lib => xautodl}/utils/gpu_manager.py | 0 {lib => xautodl}/utils/hash_utils.py | 0 {lib => xautodl}/utils/nas_utils.py | 0 {lib => xautodl}/utils/qlib_utils.py | 0 {lib => xautodl}/utils/str_utils.py | 0 {lib => xautodl}/utils/temp_sync.py | 0 {lib => xautodl}/utils/weight_watcher.py | 0 {lib => xautodl}/xlayers/__init__.py | 0 {lib => xautodl}/xlayers/drop.py | 0 {lib => xautodl}/xlayers/mlp.py | 0 .../xlayers/positional_embedding.py | 0 {lib => xautodl}/xlayers/super_activations.py | 0 {lib => xautodl}/xlayers/super_attention.py | 0 {lib => xautodl}/xlayers/super_container.py | 0 {lib => xautodl}/xlayers/super_core.py | 0 {lib => xautodl}/xlayers/super_dropout.py | 0 {lib => xautodl}/xlayers/super_linear.py | 0 {lib => xautodl}/xlayers/super_module.py | 0 {lib => xautodl}/xlayers/super_norm.py | 0 .../xlayers/super_positional_embedding.py | 0 {lib => xautodl}/xlayers/super_trade_stem.py | 0 {lib => xautodl}/xlayers/super_transformer.py | 0 {lib => xautodl}/xlayers/super_utils.py | 0 {lib => xautodl}/xlayers/weight_init.py | 0 149 files changed, 94 insertions(+), 21 deletions(-) create mode 160000 .latent-data/NAS-Bench-201 create mode 120000 lib create mode 100644 setup.py create mode 100644 xautodl/__init__.py rename {lib => xautodl}/config_utils/__init__.py (100%) rename {lib => xautodl}/config_utils/args_utils.py (100%) rename {lib => xautodl}/config_utils/attention_args.py (100%) rename {lib => xautodl}/config_utils/basic_args.py (100%) rename {lib => xautodl}/config_utils/cls_init_args.py (100%) rename {lib => xautodl}/config_utils/cls_kd_args.py (100%) rename {lib => xautodl}/config_utils/config_utils.py (100%) rename {lib => xautodl}/config_utils/pruning_args.py (100%) rename {lib => xautodl}/config_utils/random_baseline.py (100%) rename {lib => xautodl}/config_utils/search_args.py (100%) rename {lib => xautodl}/config_utils/search_single_args.py (100%) rename {lib => xautodl}/config_utils/share_args.py (100%) rename {lib => xautodl}/datasets/DownsampledImageNet.py (100%) rename {lib => xautodl}/datasets/LandmarkDataset.py (100%) rename {lib => xautodl}/datasets/SearchDatasetWrap.py (100%) rename {lib => xautodl}/datasets/__init__.py (100%) rename {lib => xautodl}/datasets/get_dataset_with_transform.py (100%) rename {lib => xautodl}/datasets/landmark_utils/__init__.py (100%) rename {lib => xautodl}/datasets/landmark_utils/point_meta.py (100%) rename {lib => xautodl}/datasets/math_adv_funcs.py (100%) rename {lib => xautodl}/datasets/math_base_funcs.py (100%) rename {lib => xautodl}/datasets/math_core.py (100%) rename {lib => xautodl}/datasets/math_dynamic_funcs.py (100%) rename {lib => xautodl}/datasets/synthetic_core.py (100%) rename {lib => xautodl}/datasets/synthetic_env.py (100%) rename {lib => xautodl}/datasets/synthetic_example.py (100%) rename {lib => xautodl}/datasets/synthetic_utils.py (100%) rename {lib => xautodl}/datasets/test_utils.py (100%) rename {lib => xautodl}/log_utils/__init__.py (100%) rename {lib => xautodl}/log_utils/logger.py (100%) rename {lib => xautodl}/log_utils/meter.py (100%) rename {lib => xautodl}/log_utils/pickle_wrap.py (100%) rename {lib => xautodl}/log_utils/time_utils.py (100%) rename {lib => xautodl}/models/CifarDenseNet.py (100%) rename {lib => xautodl}/models/CifarResNet.py (100%) rename {lib => xautodl}/models/CifarWideResNet.py (100%) rename {lib => xautodl}/models/ImageNet_MobileNetV2.py (100%) rename {lib => xautodl}/models/ImageNet_ResNet.py (100%) rename {lib => xautodl}/models/SharedUtils.py (100%) rename {lib => xautodl}/models/__init__.py (100%) rename {lib => xautodl}/models/cell_infers/__init__.py (100%) rename {lib => xautodl}/models/cell_infers/cells.py (100%) rename {lib => xautodl}/models/cell_infers/nasnet_cifar.py (100%) rename {lib => xautodl}/models/cell_infers/tiny_network.py (100%) rename {lib => xautodl}/models/cell_operations.py (100%) rename {lib => xautodl}/models/cell_searchs/__init__.py (100%) rename {lib => xautodl}/models/cell_searchs/_test_module.py (100%) rename {lib => xautodl}/models/cell_searchs/generic_model.py (100%) rename {lib => xautodl}/models/cell_searchs/genotypes.py (100%) rename {lib => xautodl}/models/cell_searchs/search_cells.py (100%) rename {lib => xautodl}/models/cell_searchs/search_model_darts.py (100%) rename {lib => xautodl}/models/cell_searchs/search_model_darts_nasnet.py (100%) rename {lib => xautodl}/models/cell_searchs/search_model_enas.py (100%) rename {lib => xautodl}/models/cell_searchs/search_model_enas_utils.py (100%) rename {lib => xautodl}/models/cell_searchs/search_model_gdas.py (100%) rename {lib => xautodl}/models/cell_searchs/search_model_gdas_frc_nasnet.py (100%) rename {lib => xautodl}/models/cell_searchs/search_model_gdas_nasnet.py (100%) rename {lib => xautodl}/models/cell_searchs/search_model_random.py (100%) rename {lib => xautodl}/models/cell_searchs/search_model_setn.py (100%) rename {lib => xautodl}/models/cell_searchs/search_model_setn_nasnet.py (100%) rename {lib => xautodl}/models/clone_weights.py (100%) rename {lib => xautodl}/models/initialization.py (100%) rename {lib => xautodl}/models/shape_infers/InferCifarResNet.py (100%) rename {lib => xautodl}/models/shape_infers/InferCifarResNet_depth.py (100%) rename {lib => xautodl}/models/shape_infers/InferCifarResNet_width.py (100%) rename {lib => xautodl}/models/shape_infers/InferImagenetResNet.py (100%) rename {lib => xautodl}/models/shape_infers/InferMobileNetV2.py (100%) rename {lib => xautodl}/models/shape_infers/InferTinyCellNet.py (100%) rename {lib => xautodl}/models/shape_infers/__init__.py (100%) rename {lib => xautodl}/models/shape_infers/shared_utils.py (100%) rename {lib => xautodl}/models/shape_searchs/SearchCifarResNet.py (100%) rename {lib => xautodl}/models/shape_searchs/SearchCifarResNet_depth.py (100%) rename {lib => xautodl}/models/shape_searchs/SearchCifarResNet_width.py (100%) rename {lib => xautodl}/models/shape_searchs/SearchImagenetResNet.py (100%) rename {lib => xautodl}/models/shape_searchs/SearchSimResNet_width.py (100%) rename {lib => xautodl}/models/shape_searchs/SoftSelect.py (100%) rename {lib => xautodl}/models/shape_searchs/__init__.py (100%) rename {lib => xautodl}/models/shape_searchs/generic_size_tiny_cell_model.py (100%) rename {lib => xautodl}/models/shape_searchs/test.py (100%) rename {lib => xautodl}/models/xcore.py (100%) rename {lib => xautodl}/nas_201_api/__init__.py (100%) rename {lib => xautodl}/nas_201_api/api_201.py (100%) rename {lib => xautodl}/nas_201_api/api_utils.py (100%) rename {lib => xautodl}/nas_infer_model/DXYs/CifarNet.py (100%) rename {lib => xautodl}/nas_infer_model/DXYs/ImageNet.py (100%) rename {lib => xautodl}/nas_infer_model/DXYs/__init__.py (100%) rename {lib => xautodl}/nas_infer_model/DXYs/base_cells.py (100%) rename {lib => xautodl}/nas_infer_model/DXYs/construct_utils.py (100%) rename {lib => xautodl}/nas_infer_model/DXYs/genotypes.py (100%) rename {lib => xautodl}/nas_infer_model/DXYs/head_utils.py (100%) rename {lib => xautodl}/nas_infer_model/__init__.py (100%) rename {lib => xautodl}/nas_infer_model/operations.py (100%) rename {lib => xautodl}/procedures/__init__.py (100%) rename {lib => xautodl}/procedures/advanced_main.py (100%) rename {lib => xautodl}/procedures/basic_main.py (100%) rename {lib => xautodl}/procedures/eval_funcs.py (100%) rename {lib => xautodl}/procedures/funcs_nasbench.py (100%) rename {lib => xautodl}/procedures/metric_utils.py (100%) rename {lib => xautodl}/procedures/optimizers.py (100%) rename {lib => xautodl}/procedures/q_exps.py (100%) rename {lib => xautodl}/procedures/search_main.py (100%) rename {lib => xautodl}/procedures/search_main_v2.py (100%) rename {lib => xautodl}/procedures/simple_KD_main.py (100%) rename {lib => xautodl}/procedures/starts.py (100%) rename {lib => xautodl}/spaces/__init__.py (100%) rename {lib => xautodl}/spaces/basic_op.py (100%) rename {lib => xautodl}/spaces/basic_space.py (100%) rename {lib => xautodl}/trade_models/__init__.py (100%) rename {lib => xautodl}/trade_models/naive_v1_model.py (100%) rename {lib => xautodl}/trade_models/naive_v2_model.py (100%) rename {lib => xautodl}/trade_models/quant_transformer.py (100%) rename {lib => xautodl}/trade_models/transformers.py (100%) rename {lib => xautodl}/utils/__init__.py (100%) rename {lib => xautodl}/utils/affine_utils.py (100%) rename {lib => xautodl}/utils/evaluation_utils.py (100%) rename {lib => xautodl}/utils/flop_benchmark.py (100%) rename {lib => xautodl}/utils/gpu_manager.py (100%) rename {lib => xautodl}/utils/hash_utils.py (100%) rename {lib => xautodl}/utils/nas_utils.py (100%) rename {lib => xautodl}/utils/qlib_utils.py (100%) rename {lib => xautodl}/utils/str_utils.py (100%) rename {lib => xautodl}/utils/temp_sync.py (100%) rename {lib => xautodl}/utils/weight_watcher.py (100%) rename {lib => xautodl}/xlayers/__init__.py (100%) rename {lib => xautodl}/xlayers/drop.py (100%) rename {lib => xautodl}/xlayers/mlp.py (100%) rename {lib => xautodl}/xlayers/positional_embedding.py (100%) rename {lib => xautodl}/xlayers/super_activations.py (100%) rename {lib => xautodl}/xlayers/super_attention.py (100%) rename {lib => xautodl}/xlayers/super_container.py (100%) rename {lib => xautodl}/xlayers/super_core.py (100%) rename {lib => xautodl}/xlayers/super_dropout.py (100%) rename {lib => xautodl}/xlayers/super_linear.py (100%) rename {lib => xautodl}/xlayers/super_module.py (100%) rename {lib => xautodl}/xlayers/super_norm.py (100%) rename {lib => xautodl}/xlayers/super_positional_embedding.py (100%) rename {lib => xautodl}/xlayers/super_trade_stem.py (100%) rename {lib => xautodl}/xlayers/super_transformer.py (100%) rename {lib => xautodl}/xlayers/super_utils.py (100%) rename {lib => xautodl}/xlayers/weight_init.py (100%) diff --git a/.gitmodules b/.gitmodules index a9f9dad..051ac47 100644 --- a/.gitmodules +++ b/.gitmodules @@ -4,3 +4,6 @@ [submodule ".latent-data/NATS-Bench"] path = .latent-data/NATS-Bench url = git@github.com:D-X-Y/NATS-Bench.git +[submodule ".latent-data/NAS-Bench-201"] + path = .latent-data/NAS-Bench-201 + url = git@github.com:D-X-Y/NAS-Bench-201.git diff --git a/.latent-data/NAS-Bench-201 b/.latent-data/NAS-Bench-201 new file mode 160000 index 0000000..8558547 --- /dev/null +++ b/.latent-data/NAS-Bench-201 @@ -0,0 +1 @@ +Subproject commit 8558547969c131f75af2725869ff1ece98e98f23 diff --git a/CHANGE-LOG.md b/CHANGE-LOG.md index e19e656..e743d50 100644 --- a/CHANGE-LOG.md +++ b/CHANGE-LOG.md @@ -6,5 +6,6 @@ - [2019.01.31] [13e908f] GDAS codes were publicly released. - [2020.07.01] [a45808b] Upgrade NAS-API to the 2.0 version. - [2020.09.16] [7052265] Create NATS-BENCH. -- [2020.10.15] [446262a] Update NATS-BENCH to version 1.0 -- [2020.12.20] [dae387a] Update NATS-BENCH to version 1.1 +- [2020.10.15] [446262a](https://github.com/D-X-Y/AutoDL-Projects/tree/446262a) Update NATS-BENCH to version 1.0 +- [2020.12.20] [dae387a](https://github.com/D-X-Y/AutoDL-Projects/tree/dae387a) Update NATS-BENCH to version 1.1 +- [2021.05.18] [98fadf8](https://github.com/D-X-Y/AutoDL-Projects/tree/98fadf8086727f6311d477d0e4f2d5bb7e41ca59) Before moving to `xautodl` diff --git a/docs/README_CN.md b/docs/README_CN.md index eb7e216..fc34443 100644 --- a/docs/README_CN.md +++ b/docs/README_CN.md @@ -88,7 +88,7 @@ ## 准备工作 -请使用`3.6`以上的`Python`，更多的Python包参见[requirements.txt](docs/requirements.txt). +请使用`3.6`以上的`Python`，更多的Python包参见[requirements.txt](requirements.txt). 请下载并且解压缩`CIFAR`和`ImageNet`到`$TORCH_HOME`. diff --git a/docs/requirements.txt b/docs/requirements.txt index b91c23a..8724d60 100644 --- a/docs/requirements.txt +++ b/docs/requirements.txt @@ -1,5 +1,5 @@ -nats_bench>=1.4 -torchvision torch -# Optional -opencv +torchvision + +nats_bench>=1.4 +nas_201_api diff --git a/exps/LFNA/lfna.py b/exps/LFNA/lfna.py index 0a34f4c..189044e 100644 --- a/exps/LFNA/lfna.py +++ b/exps/LFNA/lfna.py @@ -10,23 +10,25 @@ from tqdm import tqdm from copy import deepcopy from pathlib import Path -lib_dir = (Path(__file__).parent / ".." / ".." / "lib").resolve() -if str(lib_dir) not in sys.path: - sys.path.insert(0, str(lib_dir)) -from procedures import prepare_seed, prepare_logger, save_checkpoint, copy_checkpoint -from log_utils import time_string -from log_utils import AverageMeter, convert_secs2time +from xautodl.procedures import ( + prepare_seed, + prepare_logger, + save_checkpoint, + copy_checkpoint, +) +from xautodl.log_utils import time_string +from xautodl.log_utils import AverageMeter, convert_secs2time -from utils import split_str2indexes +from xautodl.utils import split_str2indexes -from procedures.advanced_main import basic_train_fn, basic_eval_fn -from procedures.metric_utils import SaveMetric, MSEMetric, ComposeMetric -from datasets.synthetic_core import get_synthetic_env, EnvSampler -from models.xcore import get_model -from xlayers import super_core, trunc_normal_ +from xautodl.procedures.advanced_main import basic_train_fn, basic_eval_fn +from xautodl.procedures.metric_utils import SaveMetric, MSEMetric, ComposeMetric +from xautodl.datasets.synthetic_core import get_synthetic_env, EnvSampler +from xautodl.models.xcore import get_model +from xautodl.xlayers import super_core, trunc_normal_ -from lfna_utils import lfna_setup, train_model, TimeData -from lfna_meta_model import LFNA_Meta +from xautodl.lfna_utils import lfna_setup, train_model, TimeData +from xautodl.lfna_meta_model import LFNA_Meta def epoch_train(loader, meta_model, base_model, optimizer, criterion, device, logger): diff --git a/lib b/lib new file mode 120000 index 0000000..fccce11 --- /dev/null +++ b/lib @@ -0,0 +1 @@ +xautodl \ No newline at end of file diff --git a/setup.py b/setup.py new file mode 100644 index 0000000..8c7e761 --- /dev/null +++ b/setup.py @@ -0,0 +1,59 @@ +##################################################### +# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2021.05 # +##################################################### +"""The setup function for pypi.""" +# The following is to make nats_bench avaliable on Python Package Index (PyPI) +# +# conda install -c conda-forge twine # Use twine to upload nats_bench to pypi +# +# python setup.py sdist bdist_wheel +# python setup.py --help-commands +# twine check dist/* +# +# twine upload --repository-url https://test.pypi.org/legacy/ dist/* +# twine upload dist/* +# https://pypi.org/project/xautodl +# +# TODO(xuanyidong): upload it to conda +# +# [2021.05.18] v1.0 +import os +from setuptools import setup + +NAME = "xautodl" +REQUIRES_PYTHON = ">=3.6" +DESCRIPTION = "Automated Deep Learning Package" + +VERSION = "0.9.9" + + +def read(fname="README.md"): + with open(os.path.join(os.path.dirname(__file__), fname), encoding="utf-8") as cfile: + return cfile.read() + + +# What packages are required for this module to be executed? +REQUIRED = ["numpy>=1.16.5,<=1.19.5"] + +setup( + name=NAME, + version=VERSION, + author="Xuanyi Dong", + author_email="dongxuanyi888@gmail.com", + description=DESCRIPTION, + license="MIT Licence", + keywords="NAS Dataset API DeepLearning", + url="https://github.com/D-X-Y/AutoDL-Projects", + packages=["xautodl"], + install_requires=REQUIRED, + python_requires=REQUIRES_PYTHON, + long_description=read("README.md"), + long_description_content_type="text/markdown", + classifiers=[ + "Programming Language :: Python", + "Programming Language :: Python :: 3", + "Topic :: Database", + "Topic :: Scientific/Engineering :: Artificial Intelligence", + "License :: OSI Approved :: MIT License", + ], +) diff --git a/xautodl/__init__.py b/xautodl/__init__.py new file mode 100644 index 0000000..6dabafb --- /dev/null +++ b/xautodl/__init__.py @@ -0,0 +1,6 @@ +##################################################### +# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2021.05 # +##################################################### +# An Automated Deep Learning Package to support # +# research activities. # +##################################################### diff --git a/lib/config_utils/__init__.py b/xautodl/config_utils/__init__.py similarity index 100% rename from lib/config_utils/__init__.py rename to xautodl/config_utils/__init__.py diff --git a/lib/config_utils/args_utils.py b/xautodl/config_utils/args_utils.py similarity index 100% rename from lib/config_utils/args_utils.py rename to xautodl/config_utils/args_utils.py diff --git a/lib/config_utils/attention_args.py b/xautodl/config_utils/attention_args.py similarity index 100% rename from lib/config_utils/attention_args.py rename to xautodl/config_utils/attention_args.py diff --git a/lib/config_utils/basic_args.py b/xautodl/config_utils/basic_args.py similarity index 100% rename from lib/config_utils/basic_args.py rename to xautodl/config_utils/basic_args.py diff --git a/lib/config_utils/cls_init_args.py b/xautodl/config_utils/cls_init_args.py similarity index 100% rename from lib/config_utils/cls_init_args.py rename to xautodl/config_utils/cls_init_args.py diff --git a/lib/config_utils/cls_kd_args.py b/xautodl/config_utils/cls_kd_args.py similarity index 100% rename from lib/config_utils/cls_kd_args.py rename to xautodl/config_utils/cls_kd_args.py diff --git a/lib/config_utils/config_utils.py b/xautodl/config_utils/config_utils.py similarity index 100% rename from lib/config_utils/config_utils.py rename to xautodl/config_utils/config_utils.py diff --git a/lib/config_utils/pruning_args.py b/xautodl/config_utils/pruning_args.py similarity index 100% rename from lib/config_utils/pruning_args.py rename to xautodl/config_utils/pruning_args.py diff --git a/lib/config_utils/random_baseline.py b/xautodl/config_utils/random_baseline.py similarity index 100% rename from lib/config_utils/random_baseline.py rename to xautodl/config_utils/random_baseline.py diff --git a/lib/config_utils/search_args.py b/xautodl/config_utils/search_args.py similarity index 100% rename from lib/config_utils/search_args.py rename to xautodl/config_utils/search_args.py diff --git a/lib/config_utils/search_single_args.py b/xautodl/config_utils/search_single_args.py similarity index 100% rename from lib/config_utils/search_single_args.py rename to xautodl/config_utils/search_single_args.py diff --git a/lib/config_utils/share_args.py b/xautodl/config_utils/share_args.py similarity index 100% rename from lib/config_utils/share_args.py rename to xautodl/config_utils/share_args.py diff --git a/lib/datasets/DownsampledImageNet.py b/xautodl/datasets/DownsampledImageNet.py similarity index 100% rename from lib/datasets/DownsampledImageNet.py rename to xautodl/datasets/DownsampledImageNet.py diff --git a/lib/datasets/LandmarkDataset.py b/xautodl/datasets/LandmarkDataset.py similarity index 100% rename from lib/datasets/LandmarkDataset.py rename to xautodl/datasets/LandmarkDataset.py diff --git a/lib/datasets/SearchDatasetWrap.py b/xautodl/datasets/SearchDatasetWrap.py similarity index 100% rename from lib/datasets/SearchDatasetWrap.py rename to xautodl/datasets/SearchDatasetWrap.py diff --git a/lib/datasets/__init__.py b/xautodl/datasets/__init__.py similarity index 100% rename from lib/datasets/__init__.py rename to xautodl/datasets/__init__.py diff --git a/lib/datasets/get_dataset_with_transform.py b/xautodl/datasets/get_dataset_with_transform.py similarity index 100% rename from lib/datasets/get_dataset_with_transform.py rename to xautodl/datasets/get_dataset_with_transform.py diff --git a/lib/datasets/landmark_utils/__init__.py b/xautodl/datasets/landmark_utils/__init__.py similarity index 100% rename from lib/datasets/landmark_utils/__init__.py rename to xautodl/datasets/landmark_utils/__init__.py diff --git a/lib/datasets/landmark_utils/point_meta.py b/xautodl/datasets/landmark_utils/point_meta.py similarity index 100% rename from lib/datasets/landmark_utils/point_meta.py rename to xautodl/datasets/landmark_utils/point_meta.py diff --git a/lib/datasets/math_adv_funcs.py b/xautodl/datasets/math_adv_funcs.py similarity index 100% rename from lib/datasets/math_adv_funcs.py rename to xautodl/datasets/math_adv_funcs.py diff --git a/lib/datasets/math_base_funcs.py b/xautodl/datasets/math_base_funcs.py similarity index 100% rename from lib/datasets/math_base_funcs.py rename to xautodl/datasets/math_base_funcs.py diff --git a/lib/datasets/math_core.py b/xautodl/datasets/math_core.py similarity index 100% rename from lib/datasets/math_core.py rename to xautodl/datasets/math_core.py diff --git a/lib/datasets/math_dynamic_funcs.py b/xautodl/datasets/math_dynamic_funcs.py similarity index 100% rename from lib/datasets/math_dynamic_funcs.py rename to xautodl/datasets/math_dynamic_funcs.py diff --git a/lib/datasets/synthetic_core.py b/xautodl/datasets/synthetic_core.py similarity index 100% rename from lib/datasets/synthetic_core.py rename to xautodl/datasets/synthetic_core.py diff --git a/lib/datasets/synthetic_env.py b/xautodl/datasets/synthetic_env.py similarity index 100% rename from lib/datasets/synthetic_env.py rename to xautodl/datasets/synthetic_env.py diff --git a/lib/datasets/synthetic_example.py b/xautodl/datasets/synthetic_example.py similarity index 100% rename from lib/datasets/synthetic_example.py rename to xautodl/datasets/synthetic_example.py diff --git a/lib/datasets/synthetic_utils.py b/xautodl/datasets/synthetic_utils.py similarity index 100% rename from lib/datasets/synthetic_utils.py rename to xautodl/datasets/synthetic_utils.py diff --git a/lib/datasets/test_utils.py b/xautodl/datasets/test_utils.py similarity index 100% rename from lib/datasets/test_utils.py rename to xautodl/datasets/test_utils.py diff --git a/lib/log_utils/__init__.py b/xautodl/log_utils/__init__.py similarity index 100% rename from lib/log_utils/__init__.py rename to xautodl/log_utils/__init__.py diff --git a/lib/log_utils/logger.py b/xautodl/log_utils/logger.py similarity index 100% rename from lib/log_utils/logger.py rename to xautodl/log_utils/logger.py diff --git a/lib/log_utils/meter.py b/xautodl/log_utils/meter.py similarity index 100% rename from lib/log_utils/meter.py rename to xautodl/log_utils/meter.py diff --git a/lib/log_utils/pickle_wrap.py b/xautodl/log_utils/pickle_wrap.py similarity index 100% rename from lib/log_utils/pickle_wrap.py rename to xautodl/log_utils/pickle_wrap.py diff --git a/lib/log_utils/time_utils.py b/xautodl/log_utils/time_utils.py similarity index 100% rename from lib/log_utils/time_utils.py rename to xautodl/log_utils/time_utils.py diff --git a/lib/models/CifarDenseNet.py b/xautodl/models/CifarDenseNet.py similarity index 100% rename from lib/models/CifarDenseNet.py rename to xautodl/models/CifarDenseNet.py diff --git a/lib/models/CifarResNet.py b/xautodl/models/CifarResNet.py similarity index 100% rename from lib/models/CifarResNet.py rename to xautodl/models/CifarResNet.py diff --git a/lib/models/CifarWideResNet.py b/xautodl/models/CifarWideResNet.py similarity index 100% rename from lib/models/CifarWideResNet.py rename to xautodl/models/CifarWideResNet.py diff --git a/lib/models/ImageNet_MobileNetV2.py b/xautodl/models/ImageNet_MobileNetV2.py similarity index 100% rename from lib/models/ImageNet_MobileNetV2.py rename to xautodl/models/ImageNet_MobileNetV2.py diff --git a/lib/models/ImageNet_ResNet.py b/xautodl/models/ImageNet_ResNet.py similarity index 100% rename from lib/models/ImageNet_ResNet.py rename to xautodl/models/ImageNet_ResNet.py diff --git a/lib/models/SharedUtils.py b/xautodl/models/SharedUtils.py similarity index 100% rename from lib/models/SharedUtils.py rename to xautodl/models/SharedUtils.py diff --git a/lib/models/__init__.py b/xautodl/models/__init__.py similarity index 100% rename from lib/models/__init__.py rename to xautodl/models/__init__.py diff --git a/lib/models/cell_infers/__init__.py b/xautodl/models/cell_infers/__init__.py similarity index 100% rename from lib/models/cell_infers/__init__.py rename to xautodl/models/cell_infers/__init__.py diff --git a/lib/models/cell_infers/cells.py b/xautodl/models/cell_infers/cells.py similarity index 100% rename from lib/models/cell_infers/cells.py rename to xautodl/models/cell_infers/cells.py diff --git a/lib/models/cell_infers/nasnet_cifar.py b/xautodl/models/cell_infers/nasnet_cifar.py similarity index 100% rename from lib/models/cell_infers/nasnet_cifar.py rename to xautodl/models/cell_infers/nasnet_cifar.py diff --git a/lib/models/cell_infers/tiny_network.py b/xautodl/models/cell_infers/tiny_network.py similarity index 100% rename from lib/models/cell_infers/tiny_network.py rename to xautodl/models/cell_infers/tiny_network.py diff --git a/lib/models/cell_operations.py b/xautodl/models/cell_operations.py similarity index 100% rename from lib/models/cell_operations.py rename to xautodl/models/cell_operations.py diff --git a/lib/models/cell_searchs/__init__.py b/xautodl/models/cell_searchs/__init__.py similarity index 100% rename from lib/models/cell_searchs/__init__.py rename to xautodl/models/cell_searchs/__init__.py diff --git a/lib/models/cell_searchs/_test_module.py b/xautodl/models/cell_searchs/_test_module.py similarity index 100% rename from lib/models/cell_searchs/_test_module.py rename to xautodl/models/cell_searchs/_test_module.py diff --git a/lib/models/cell_searchs/generic_model.py b/xautodl/models/cell_searchs/generic_model.py similarity index 100% rename from lib/models/cell_searchs/generic_model.py rename to xautodl/models/cell_searchs/generic_model.py diff --git a/lib/models/cell_searchs/genotypes.py b/xautodl/models/cell_searchs/genotypes.py similarity index 100% rename from lib/models/cell_searchs/genotypes.py rename to xautodl/models/cell_searchs/genotypes.py diff --git a/lib/models/cell_searchs/search_cells.py b/xautodl/models/cell_searchs/search_cells.py similarity index 100% rename from lib/models/cell_searchs/search_cells.py rename to xautodl/models/cell_searchs/search_cells.py diff --git a/lib/models/cell_searchs/search_model_darts.py b/xautodl/models/cell_searchs/search_model_darts.py similarity index 100% rename from lib/models/cell_searchs/search_model_darts.py rename to xautodl/models/cell_searchs/search_model_darts.py diff --git a/lib/models/cell_searchs/search_model_darts_nasnet.py b/xautodl/models/cell_searchs/search_model_darts_nasnet.py similarity index 100% rename from lib/models/cell_searchs/search_model_darts_nasnet.py rename to xautodl/models/cell_searchs/search_model_darts_nasnet.py diff --git a/lib/models/cell_searchs/search_model_enas.py b/xautodl/models/cell_searchs/search_model_enas.py similarity index 100% rename from lib/models/cell_searchs/search_model_enas.py rename to xautodl/models/cell_searchs/search_model_enas.py diff --git a/lib/models/cell_searchs/search_model_enas_utils.py b/xautodl/models/cell_searchs/search_model_enas_utils.py similarity index 100% rename from lib/models/cell_searchs/search_model_enas_utils.py rename to xautodl/models/cell_searchs/search_model_enas_utils.py diff --git a/lib/models/cell_searchs/search_model_gdas.py b/xautodl/models/cell_searchs/search_model_gdas.py similarity index 100% rename from lib/models/cell_searchs/search_model_gdas.py rename to xautodl/models/cell_searchs/search_model_gdas.py diff --git a/lib/models/cell_searchs/search_model_gdas_frc_nasnet.py b/xautodl/models/cell_searchs/search_model_gdas_frc_nasnet.py similarity index 100% rename from lib/models/cell_searchs/search_model_gdas_frc_nasnet.py rename to xautodl/models/cell_searchs/search_model_gdas_frc_nasnet.py diff --git a/lib/models/cell_searchs/search_model_gdas_nasnet.py b/xautodl/models/cell_searchs/search_model_gdas_nasnet.py similarity index 100% rename from lib/models/cell_searchs/search_model_gdas_nasnet.py rename to xautodl/models/cell_searchs/search_model_gdas_nasnet.py diff --git a/lib/models/cell_searchs/search_model_random.py b/xautodl/models/cell_searchs/search_model_random.py similarity index 100% rename from lib/models/cell_searchs/search_model_random.py rename to xautodl/models/cell_searchs/search_model_random.py diff --git a/lib/models/cell_searchs/search_model_setn.py b/xautodl/models/cell_searchs/search_model_setn.py similarity index 100% rename from lib/models/cell_searchs/search_model_setn.py rename to xautodl/models/cell_searchs/search_model_setn.py diff --git a/lib/models/cell_searchs/search_model_setn_nasnet.py b/xautodl/models/cell_searchs/search_model_setn_nasnet.py similarity index 100% rename from lib/models/cell_searchs/search_model_setn_nasnet.py rename to xautodl/models/cell_searchs/search_model_setn_nasnet.py diff --git a/lib/models/clone_weights.py b/xautodl/models/clone_weights.py similarity index 100% rename from lib/models/clone_weights.py rename to xautodl/models/clone_weights.py diff --git a/lib/models/initialization.py b/xautodl/models/initialization.py similarity index 100% rename from lib/models/initialization.py rename to xautodl/models/initialization.py diff --git a/lib/models/shape_infers/InferCifarResNet.py b/xautodl/models/shape_infers/InferCifarResNet.py similarity index 100% rename from lib/models/shape_infers/InferCifarResNet.py rename to xautodl/models/shape_infers/InferCifarResNet.py diff --git a/lib/models/shape_infers/InferCifarResNet_depth.py b/xautodl/models/shape_infers/InferCifarResNet_depth.py similarity index 100% rename from lib/models/shape_infers/InferCifarResNet_depth.py rename to xautodl/models/shape_infers/InferCifarResNet_depth.py diff --git a/lib/models/shape_infers/InferCifarResNet_width.py b/xautodl/models/shape_infers/InferCifarResNet_width.py similarity index 100% rename from lib/models/shape_infers/InferCifarResNet_width.py rename to xautodl/models/shape_infers/InferCifarResNet_width.py diff --git a/lib/models/shape_infers/InferImagenetResNet.py b/xautodl/models/shape_infers/InferImagenetResNet.py similarity index 100% rename from lib/models/shape_infers/InferImagenetResNet.py rename to xautodl/models/shape_infers/InferImagenetResNet.py diff --git a/lib/models/shape_infers/InferMobileNetV2.py b/xautodl/models/shape_infers/InferMobileNetV2.py similarity index 100% rename from lib/models/shape_infers/InferMobileNetV2.py rename to xautodl/models/shape_infers/InferMobileNetV2.py diff --git a/lib/models/shape_infers/InferTinyCellNet.py b/xautodl/models/shape_infers/InferTinyCellNet.py similarity index 100% rename from lib/models/shape_infers/InferTinyCellNet.py rename to xautodl/models/shape_infers/InferTinyCellNet.py diff --git a/lib/models/shape_infers/__init__.py b/xautodl/models/shape_infers/__init__.py similarity index 100% rename from lib/models/shape_infers/__init__.py rename to xautodl/models/shape_infers/__init__.py diff --git a/lib/models/shape_infers/shared_utils.py b/xautodl/models/shape_infers/shared_utils.py similarity index 100% rename from lib/models/shape_infers/shared_utils.py rename to xautodl/models/shape_infers/shared_utils.py diff --git a/lib/models/shape_searchs/SearchCifarResNet.py b/xautodl/models/shape_searchs/SearchCifarResNet.py similarity index 100% rename from lib/models/shape_searchs/SearchCifarResNet.py rename to xautodl/models/shape_searchs/SearchCifarResNet.py diff --git a/lib/models/shape_searchs/SearchCifarResNet_depth.py b/xautodl/models/shape_searchs/SearchCifarResNet_depth.py similarity index 100% rename from lib/models/shape_searchs/SearchCifarResNet_depth.py rename to xautodl/models/shape_searchs/SearchCifarResNet_depth.py diff --git a/lib/models/shape_searchs/SearchCifarResNet_width.py b/xautodl/models/shape_searchs/SearchCifarResNet_width.py similarity index 100% rename from lib/models/shape_searchs/SearchCifarResNet_width.py rename to xautodl/models/shape_searchs/SearchCifarResNet_width.py diff --git a/lib/models/shape_searchs/SearchImagenetResNet.py b/xautodl/models/shape_searchs/SearchImagenetResNet.py similarity index 100% rename from lib/models/shape_searchs/SearchImagenetResNet.py rename to xautodl/models/shape_searchs/SearchImagenetResNet.py diff --git a/lib/models/shape_searchs/SearchSimResNet_width.py b/xautodl/models/shape_searchs/SearchSimResNet_width.py similarity index 100% rename from lib/models/shape_searchs/SearchSimResNet_width.py rename to xautodl/models/shape_searchs/SearchSimResNet_width.py diff --git a/lib/models/shape_searchs/SoftSelect.py b/xautodl/models/shape_searchs/SoftSelect.py similarity index 100% rename from lib/models/shape_searchs/SoftSelect.py rename to xautodl/models/shape_searchs/SoftSelect.py diff --git a/lib/models/shape_searchs/__init__.py b/xautodl/models/shape_searchs/__init__.py similarity index 100% rename from lib/models/shape_searchs/__init__.py rename to xautodl/models/shape_searchs/__init__.py diff --git a/lib/models/shape_searchs/generic_size_tiny_cell_model.py b/xautodl/models/shape_searchs/generic_size_tiny_cell_model.py similarity index 100% rename from lib/models/shape_searchs/generic_size_tiny_cell_model.py rename to xautodl/models/shape_searchs/generic_size_tiny_cell_model.py diff --git a/lib/models/shape_searchs/test.py b/xautodl/models/shape_searchs/test.py similarity index 100% rename from lib/models/shape_searchs/test.py rename to xautodl/models/shape_searchs/test.py diff --git a/lib/models/xcore.py b/xautodl/models/xcore.py similarity index 100% rename from lib/models/xcore.py rename to xautodl/models/xcore.py diff --git a/lib/nas_201_api/__init__.py b/xautodl/nas_201_api/__init__.py similarity index 100% rename from lib/nas_201_api/__init__.py rename to xautodl/nas_201_api/__init__.py diff --git a/lib/nas_201_api/api_201.py b/xautodl/nas_201_api/api_201.py similarity index 100% rename from lib/nas_201_api/api_201.py rename to xautodl/nas_201_api/api_201.py diff --git a/lib/nas_201_api/api_utils.py b/xautodl/nas_201_api/api_utils.py similarity index 100% rename from lib/nas_201_api/api_utils.py rename to xautodl/nas_201_api/api_utils.py diff --git a/lib/nas_infer_model/DXYs/CifarNet.py b/xautodl/nas_infer_model/DXYs/CifarNet.py similarity index 100% rename from lib/nas_infer_model/DXYs/CifarNet.py rename to xautodl/nas_infer_model/DXYs/CifarNet.py diff --git a/lib/nas_infer_model/DXYs/ImageNet.py b/xautodl/nas_infer_model/DXYs/ImageNet.py similarity index 100% rename from lib/nas_infer_model/DXYs/ImageNet.py rename to xautodl/nas_infer_model/DXYs/ImageNet.py diff --git a/lib/nas_infer_model/DXYs/__init__.py b/xautodl/nas_infer_model/DXYs/__init__.py similarity index 100% rename from lib/nas_infer_model/DXYs/__init__.py rename to xautodl/nas_infer_model/DXYs/__init__.py diff --git a/lib/nas_infer_model/DXYs/base_cells.py b/xautodl/nas_infer_model/DXYs/base_cells.py similarity index 100% rename from lib/nas_infer_model/DXYs/base_cells.py rename to xautodl/nas_infer_model/DXYs/base_cells.py diff --git a/lib/nas_infer_model/DXYs/construct_utils.py b/xautodl/nas_infer_model/DXYs/construct_utils.py similarity index 100% rename from lib/nas_infer_model/DXYs/construct_utils.py rename to xautodl/nas_infer_model/DXYs/construct_utils.py diff --git a/lib/nas_infer_model/DXYs/genotypes.py b/xautodl/nas_infer_model/DXYs/genotypes.py similarity index 100% rename from lib/nas_infer_model/DXYs/genotypes.py rename to xautodl/nas_infer_model/DXYs/genotypes.py diff --git a/lib/nas_infer_model/DXYs/head_utils.py b/xautodl/nas_infer_model/DXYs/head_utils.py similarity index 100% rename from lib/nas_infer_model/DXYs/head_utils.py rename to xautodl/nas_infer_model/DXYs/head_utils.py diff --git a/lib/nas_infer_model/__init__.py b/xautodl/nas_infer_model/__init__.py similarity index 100% rename from lib/nas_infer_model/__init__.py rename to xautodl/nas_infer_model/__init__.py diff --git a/lib/nas_infer_model/operations.py b/xautodl/nas_infer_model/operations.py similarity index 100% rename from lib/nas_infer_model/operations.py rename to xautodl/nas_infer_model/operations.py diff --git a/lib/procedures/__init__.py b/xautodl/procedures/__init__.py similarity index 100% rename from lib/procedures/__init__.py rename to xautodl/procedures/__init__.py diff --git a/lib/procedures/advanced_main.py b/xautodl/procedures/advanced_main.py similarity index 100% rename from lib/procedures/advanced_main.py rename to xautodl/procedures/advanced_main.py diff --git a/lib/procedures/basic_main.py b/xautodl/procedures/basic_main.py similarity index 100% rename from lib/procedures/basic_main.py rename to xautodl/procedures/basic_main.py diff --git a/lib/procedures/eval_funcs.py b/xautodl/procedures/eval_funcs.py similarity index 100% rename from lib/procedures/eval_funcs.py rename to xautodl/procedures/eval_funcs.py diff --git a/lib/procedures/funcs_nasbench.py b/xautodl/procedures/funcs_nasbench.py similarity index 100% rename from lib/procedures/funcs_nasbench.py rename to xautodl/procedures/funcs_nasbench.py diff --git a/lib/procedures/metric_utils.py b/xautodl/procedures/metric_utils.py similarity index 100% rename from lib/procedures/metric_utils.py rename to xautodl/procedures/metric_utils.py diff --git a/lib/procedures/optimizers.py b/xautodl/procedures/optimizers.py similarity index 100% rename from lib/procedures/optimizers.py rename to xautodl/procedures/optimizers.py diff --git a/lib/procedures/q_exps.py b/xautodl/procedures/q_exps.py similarity index 100% rename from lib/procedures/q_exps.py rename to xautodl/procedures/q_exps.py diff --git a/lib/procedures/search_main.py b/xautodl/procedures/search_main.py similarity index 100% rename from lib/procedures/search_main.py rename to xautodl/procedures/search_main.py diff --git a/lib/procedures/search_main_v2.py b/xautodl/procedures/search_main_v2.py similarity index 100% rename from lib/procedures/search_main_v2.py rename to xautodl/procedures/search_main_v2.py diff --git a/lib/procedures/simple_KD_main.py b/xautodl/procedures/simple_KD_main.py similarity index 100% rename from lib/procedures/simple_KD_main.py rename to xautodl/procedures/simple_KD_main.py diff --git a/lib/procedures/starts.py b/xautodl/procedures/starts.py similarity index 100% rename from lib/procedures/starts.py rename to xautodl/procedures/starts.py diff --git a/lib/spaces/__init__.py b/xautodl/spaces/__init__.py similarity index 100% rename from lib/spaces/__init__.py rename to xautodl/spaces/__init__.py diff --git a/lib/spaces/basic_op.py b/xautodl/spaces/basic_op.py similarity index 100% rename from lib/spaces/basic_op.py rename to xautodl/spaces/basic_op.py diff --git a/lib/spaces/basic_space.py b/xautodl/spaces/basic_space.py similarity index 100% rename from lib/spaces/basic_space.py rename to xautodl/spaces/basic_space.py diff --git a/lib/trade_models/__init__.py b/xautodl/trade_models/__init__.py similarity index 100% rename from lib/trade_models/__init__.py rename to xautodl/trade_models/__init__.py diff --git a/lib/trade_models/naive_v1_model.py b/xautodl/trade_models/naive_v1_model.py similarity index 100% rename from lib/trade_models/naive_v1_model.py rename to xautodl/trade_models/naive_v1_model.py diff --git a/lib/trade_models/naive_v2_model.py b/xautodl/trade_models/naive_v2_model.py similarity index 100% rename from lib/trade_models/naive_v2_model.py rename to xautodl/trade_models/naive_v2_model.py diff --git a/lib/trade_models/quant_transformer.py b/xautodl/trade_models/quant_transformer.py similarity index 100% rename from lib/trade_models/quant_transformer.py rename to xautodl/trade_models/quant_transformer.py diff --git a/lib/trade_models/transformers.py b/xautodl/trade_models/transformers.py similarity index 100% rename from lib/trade_models/transformers.py rename to xautodl/trade_models/transformers.py diff --git a/lib/utils/__init__.py b/xautodl/utils/__init__.py similarity index 100% rename from lib/utils/__init__.py rename to xautodl/utils/__init__.py diff --git a/lib/utils/affine_utils.py b/xautodl/utils/affine_utils.py similarity index 100% rename from lib/utils/affine_utils.py rename to xautodl/utils/affine_utils.py diff --git a/lib/utils/evaluation_utils.py b/xautodl/utils/evaluation_utils.py similarity index 100% rename from lib/utils/evaluation_utils.py rename to xautodl/utils/evaluation_utils.py diff --git a/lib/utils/flop_benchmark.py b/xautodl/utils/flop_benchmark.py similarity index 100% rename from lib/utils/flop_benchmark.py rename to xautodl/utils/flop_benchmark.py diff --git a/lib/utils/gpu_manager.py b/xautodl/utils/gpu_manager.py similarity index 100% rename from lib/utils/gpu_manager.py rename to xautodl/utils/gpu_manager.py diff --git a/lib/utils/hash_utils.py b/xautodl/utils/hash_utils.py similarity index 100% rename from lib/utils/hash_utils.py rename to xautodl/utils/hash_utils.py diff --git a/lib/utils/nas_utils.py b/xautodl/utils/nas_utils.py similarity index 100% rename from lib/utils/nas_utils.py rename to xautodl/utils/nas_utils.py diff --git a/lib/utils/qlib_utils.py b/xautodl/utils/qlib_utils.py similarity index 100% rename from lib/utils/qlib_utils.py rename to xautodl/utils/qlib_utils.py diff --git a/lib/utils/str_utils.py b/xautodl/utils/str_utils.py similarity index 100% rename from lib/utils/str_utils.py rename to xautodl/utils/str_utils.py diff --git a/lib/utils/temp_sync.py b/xautodl/utils/temp_sync.py similarity index 100% rename from lib/utils/temp_sync.py rename to xautodl/utils/temp_sync.py diff --git a/lib/utils/weight_watcher.py b/xautodl/utils/weight_watcher.py similarity index 100% rename from lib/utils/weight_watcher.py rename to xautodl/utils/weight_watcher.py diff --git a/lib/xlayers/__init__.py b/xautodl/xlayers/__init__.py similarity index 100% rename from lib/xlayers/__init__.py rename to xautodl/xlayers/__init__.py diff --git a/lib/xlayers/drop.py b/xautodl/xlayers/drop.py similarity index 100% rename from lib/xlayers/drop.py rename to xautodl/xlayers/drop.py diff --git a/lib/xlayers/mlp.py b/xautodl/xlayers/mlp.py similarity index 100% rename from lib/xlayers/mlp.py rename to xautodl/xlayers/mlp.py diff --git a/lib/xlayers/positional_embedding.py b/xautodl/xlayers/positional_embedding.py similarity index 100% rename from lib/xlayers/positional_embedding.py rename to xautodl/xlayers/positional_embedding.py diff --git a/lib/xlayers/super_activations.py b/xautodl/xlayers/super_activations.py similarity index 100% rename from lib/xlayers/super_activations.py rename to xautodl/xlayers/super_activations.py diff --git a/lib/xlayers/super_attention.py b/xautodl/xlayers/super_attention.py similarity index 100% rename from lib/xlayers/super_attention.py rename to xautodl/xlayers/super_attention.py diff --git a/lib/xlayers/super_container.py b/xautodl/xlayers/super_container.py similarity index 100% rename from lib/xlayers/super_container.py rename to xautodl/xlayers/super_container.py diff --git a/lib/xlayers/super_core.py b/xautodl/xlayers/super_core.py similarity index 100% rename from lib/xlayers/super_core.py rename to xautodl/xlayers/super_core.py diff --git a/lib/xlayers/super_dropout.py b/xautodl/xlayers/super_dropout.py similarity index 100% rename from lib/xlayers/super_dropout.py rename to xautodl/xlayers/super_dropout.py diff --git a/lib/xlayers/super_linear.py b/xautodl/xlayers/super_linear.py similarity index 100% rename from lib/xlayers/super_linear.py rename to xautodl/xlayers/super_linear.py diff --git a/lib/xlayers/super_module.py b/xautodl/xlayers/super_module.py similarity index 100% rename from lib/xlayers/super_module.py rename to xautodl/xlayers/super_module.py diff --git a/lib/xlayers/super_norm.py b/xautodl/xlayers/super_norm.py similarity index 100% rename from lib/xlayers/super_norm.py rename to xautodl/xlayers/super_norm.py diff --git a/lib/xlayers/super_positional_embedding.py b/xautodl/xlayers/super_positional_embedding.py similarity index 100% rename from lib/xlayers/super_positional_embedding.py rename to xautodl/xlayers/super_positional_embedding.py diff --git a/lib/xlayers/super_trade_stem.py b/xautodl/xlayers/super_trade_stem.py similarity index 100% rename from lib/xlayers/super_trade_stem.py rename to xautodl/xlayers/super_trade_stem.py diff --git a/lib/xlayers/super_transformer.py b/xautodl/xlayers/super_transformer.py similarity index 100% rename from lib/xlayers/super_transformer.py rename to xautodl/xlayers/super_transformer.py diff --git a/lib/xlayers/super_utils.py b/xautodl/xlayers/super_utils.py similarity index 100% rename from lib/xlayers/super_utils.py rename to xautodl/xlayers/super_utils.py diff --git a/lib/xlayers/weight_init.py b/xautodl/xlayers/weight_init.py similarity index 100% rename from lib/xlayers/weight_init.py rename to xautodl/xlayers/weight_init.py