├── .gitignore ├── .gitmodules ├── LICENSE ├── MANIFEST.in ├── README.md ├── aw_nas ├── VERSION ├── __init__.py ├── base.py ├── btcs │ ├── __init__.py │ ├── enas.py │ ├── layer2 │ │ ├── __init__.py │ │ ├── bi_final_model.py │ │ ├── controller.py │ │ ├── diff_weights_manager.py │ │ ├── final_model.py │ │ ├── search_space.py │ │ └── weights_manager.py │ ├── nasbench_101.py │ ├── nasbench_201.py │ ├── nasbench_201_close.py │ ├── nasbench_301.py │ └── zero_cost_proxy.py ├── common.py ├── controller │ ├── __init__.py │ ├── base.py │ ├── cars_evo.py │ ├── dense_controller.py │ ├── differentiable.py │ ├── evo.py │ ├── ofa.py │ ├── predictor_based.py │ ├── rejection.py │ ├── rl.py │ ├── rl_agents.py │ └── rl_networks.py ├── dataset │ ├── __init__.py │ ├── base.py │ ├── bdd100k.py │ ├── cifar10.py │ ├── cifar100.py │ ├── coco.py │ ├── det_augmentation.py │ ├── det_transform.py │ ├── imagenet.py │ ├── imagenet_downsample.py │ ├── miniimagenet.py │ ├── ptb.py │ ├── svhn.py │ ├── tiny_imagenet.py │ └── voc.py ├── evaluator │ ├── __init__.py │ ├── arch_network.py │ ├── base.py │ ├── bftune.py │ ├── compare.py │ ├── few_shot_shared_weight.py │ ├── mepa.py │ ├── shared_weight.py │ └── tune.py ├── final │ ├── __init__.py │ ├── base.py │ ├── bnn_model.py │ ├── cnn_model.py │ ├── cnn_trainer.py │ ├── dense.py │ ├── det_model.py │ ├── det_trainer.py │ ├── general_model.py │ ├── ofa_model.py │ ├── rnn_model.py │ ├── rnn_trainer.py │ ├── wrapper_model.py │ └── wrapper_trainer.py ├── germ │ ├── README.md │ ├── __init__.py │ ├── decisions.py │ ├── germ.py │ ├── mbv2.py │ ├── mobiledet.py │ ├── nb201.py │ ├── necks │ │ ├── __init__.py │ │ └── fpn.py │ ├── searchable_blocks.py │ ├── shufflev2.py │ ├── supernet.py │ └── utils.py ├── hardware │ ├── __init__.py │ ├── base.py │ ├── compiler │ │ ├── __init__.py │ │ ├── dpu.py │ │ └── xavier.py │ ├── ofa_obj.py │ └── utils.py ├── main.py ├── main_hardware.py ├── objective │ ├── __init__.py │ ├── base.py │ ├── container.py │ ├── detection.py │ ├── detection_utils │ │ ├── __init__.py │ │ ├── anchors_generator.py │ │ ├── base.py │ │ ├── losses.py │ │ ├── matcher.py │ │ ├── metrics.py │ │ └── post_processing.py │ ├── fault_injection.py │ ├── flops.py │ ├── hardware.py │ ├── image.py │ ├── language.py │ ├── mmdet_head.py │ ├── ofa.py │ ├── zerocost.py │ └── zeroshot.py ├── ops │ ├── __init__.py │ ├── baseline_ops.py │ ├── bnn_ops.py │ └── ops.py ├── plugin.py ├── rollout │ ├── __init__.py │ ├── base.py │ ├── compare.py │ ├── dense.py │ ├── ensemble.py │ ├── general.py │ ├── mutation.py │ ├── ofa.py │ └── wrapper.py ├── trainer │ ├── __init__.py │ ├── async_trainer.py │ ├── base.py │ ├── meta.py │ ├── ray_dispatcher.py │ └── simple.py ├── utils │ ├── __init__.py │ ├── box_utils.py │ ├── common_utils.py │ ├── elapse_utils.py │ ├── exception.py │ ├── log.py │ ├── lr_scheduler.py │ ├── parallel_utils.py │ ├── registry.py │ ├── torch_utils.py │ └── vis_utils.py └── weights_manager │ ├── __init__.py │ ├── base.py │ ├── close_net.py │ ├── dense.py │ ├── diff_super_net.py │ ├── ensemble.py │ ├── headers │ ├── __init__.py │ ├── anchor_based.py │ ├── atss.py │ ├── bifpn.py │ └── classifiers.py │ ├── morphism.py │ ├── necks │ ├── __init__.py │ ├── base.py │ ├── bifpn.py │ ├── fpn.py │ └── ssd.py │ ├── ofa.py │ ├── ofa_backbone.py │ ├── rnn_diff_super_net.py │ ├── rnn_shared.py │ ├── rnn_super_net.py │ ├── shared.py │ ├── stub.py │ ├── super_net.py │ └── wrapper.py ├── doc ├── awnas.pdf ├── darts_flow.md ├── development.md ├── enas_cfg_notes.md ├── hardware.md └── pics │ ├── coverage.svg │ ├── enas_cell.jpg │ ├── flow.pdf │ ├── framework.pdf │ ├── hardware_profiling.pdf │ ├── logo_nicsefc.jpg │ └── logo_novauto.jpg ├── examples ├── baselines │ ├── densenet121.yaml │ ├── generate_resnet.py │ ├── mobilenet-v2.yaml │ ├── resnet18.yaml │ ├── resnet18_distributed.yaml │ ├── resnet18_imgnet.yaml │ ├── resnet20.yaml │ ├── resnet50.yaml │ └── vgg16.yaml ├── basic │ ├── darts.yaml │ ├── enas.yaml │ ├── enas_evo.yaml │ ├── enas_multigpu.yaml │ ├── final_templates │ │ ├── final_template.yaml │ │ └── sur_template.yaml │ └── pcdarts.yaml ├── datasets │ ├── imagenet │ │ ├── enas_search.yaml │ │ ├── final_template.yaml │ │ ├── mepa3_search.yaml │ │ └── sur_template.yaml │ └── tiny-imagenet │ │ └── resnet20.yaml ├── detection │ ├── README.md │ ├── VOC_ssd_final_distill.yaml │ ├── VOC_ssd_supernet_distill.yaml │ ├── mmdet_final.yaml │ ├── ssd_async_search.yaml │ ├── ssd_final.yaml │ └── ssd_supernet_training.yaml ├── fewshot │ ├── README.md │ ├── tnas_5way_1shot.yaml │ ├── tnas_5way_1shot_reptile.yaml │ └── tnas_5way_5shot.yaml ├── germ │ ├── example_code_snippet.py │ ├── example_code_snippet.py.sscfg.yaml │ ├── example_code_snippet_search.yaml │ ├── example_code_snippet_ss.yaml │ ├── mbv2_code_snippet_ss.yaml │ ├── mbv2_final.yaml │ ├── mbv2_search.yaml │ └── mbv2_test.yaml ├── hardware │ ├── configs │ │ ├── general.yaml │ │ ├── hardware_obj.yaml │ │ ├── ofa_final.yaml │ │ └── ofa_lat.yaml │ ├── det_ofa_hardware.yaml │ └── det_ofa_xavier.yaml ├── mloss │ ├── README.md │ ├── darts │ │ ├── darts_final.yaml │ │ ├── darts_search.yaml │ │ └── derived_genotypes.yaml │ ├── enas │ │ ├── derived_genotypes.yaml │ │ ├── enas_final.yaml │ │ └── enas_search.yaml │ ├── pcdarts │ │ ├── pcdarts_final.yaml │ │ ├── pcdarts_genotypes.yaml │ │ └── pcdarts_search.yaml │ └── snas │ │ ├── derived_genotypes.yaml │ │ ├── snas_final.yaml │ │ └── snas_search.yaml ├── nasbench │ ├── nasbench-101_1shot.yaml │ ├── nasbench-101_evo.yaml │ ├── nasbench-101_gates_sa.yaml │ ├── nasbench-101_sa.yaml │ ├── nasbench-201_final_template.yaml │ ├── nasbench-201_gates_sa.yaml │ ├── nasbench-301.yaml │ ├── nasbench-301_plateaulr.yaml │ ├── nasbench_201_evo.yaml │ └── nasbench_201_sa.yaml ├── ofa │ ├── cifar100_mbv3_search.yaml │ ├── mbv3_final.yaml │ ├── mbv3_pareto_search.yaml │ ├── mbv3_search.yaml │ ├── mbv3_supernet_training.yaml │ ├── mnas_final.yaml │ ├── mnas_search.yaml │ └── mnas_supernet_training.yaml ├── plugins │ ├── fbnet │ │ ├── fbnet_plugin.py │ │ └── sample_fbnet_diff.yaml │ ├── pool │ │ └── pool_plugin.py │ └── robustness │ │ ├── 1_robustness.py │ │ ├── 2_multi_evaluator.py │ │ ├── cfgs │ │ ├── example_supernet_training.yaml │ │ ├── final │ │ │ ├── train-cifar10-MSRobNet-1000.yaml │ │ │ └── train-cifar100-resnet18.yaml │ │ └── stagewise_predbased │ │ │ ├── final_ms2000_predbased.yaml │ │ │ ├── search-24c_stagewisess_pgd_gates.yaml │ │ │ ├── search-multishoteval_stagewisess_fgsm_gates_2000M.yaml │ │ │ └── search-multishoteval_stagewisess_pgd_gates_2000M.yaml │ │ ├── robustness_archembedder.py │ │ ├── robustness_final_model.py │ │ ├── robustness_ss.py │ │ ├── robustness_weights_manager.py │ │ ├── test_acc_attack.py │ │ ├── test_black_attack.py │ │ ├── test_distance_attack.py │ │ └── test_rob_ss.py └── research │ ├── bbssp │ ├── README.md │ ├── diff_config │ │ ├── diff_cifar10_allblock.yaml │ │ └── speed_diff_allblock.txt │ ├── rl_config │ │ ├── rl_cifar10_allblock.yaml │ │ └── speed_cifar_allblock.txt │ └── ssp_plugin │ │ └── ssp_plugin.py │ ├── bnn │ └── README.md │ ├── close │ ├── README.md │ ├── nb201 │ │ ├── node_cl-2468_lr-4_mlp-200x3_wit.yaml │ │ ├── node_cl-246_lr-4_mlp-200x3_wit.yaml │ │ └── node_cl-24_lr-4_mlp-200x3_wit.yaml │ ├── nb301 │ │ ├── node_cl-2468_lr-4_mlp-200x3_wit.yaml │ │ ├── node_cl-246_lr-4_mlp-200x3_wit.yaml │ │ └── node_cl-24_lr-4_mlp-200x3_wit.yaml │ ├── nds │ │ ├── resnet_cl-9-22446688_lr-4_pre.yaml │ │ └── resnexta_cl-9-2468_lr-4_pre.yaml │ ├── run_derive_ckpts.py │ ├── run_derive_epoch.py │ └── run_supernet_training.py │ ├── dele │ ├── README.md │ ├── cfgs │ │ ├── dynamic_ensemble_nb201_gates_config.yaml │ │ ├── nasbench-201_gates_search.yaml │ │ └── train_nb201_gates_config.yaml │ ├── dataset.py │ ├── dynamic_ensemble_train_predictor.py │ ├── plugin │ │ └── nds_arch_embedder.py │ ├── train_predictor.py │ └── utils.py │ ├── ftt-nas │ ├── README.md │ ├── adsaf.yaml │ ├── adsaf_final.yaml │ ├── fixed_point_plugins │ │ ├── fixed_point_patch.py │ │ ├── fixed_point_patch_new.py │ │ ├── fixed_point_rram_patch.py │ │ ├── fixed_point_rram_patch_all.py │ │ └── fixed_point_rram_patch_bit.py │ ├── mibb.yaml │ ├── run_adsaf.sh │ ├── run_adsaf_search.sh │ ├── run_mibb.sh │ ├── run_mibb_search.sh │ └── test_fix.py │ ├── gates │ ├── README.md │ ├── dump_nb101_allv.py │ ├── nbc_flow_gcn128x4.yaml │ ├── nbc_flow_gcn128x4_pairdiff.yaml │ └── print_res.sh │ ├── gibbon │ └── README.md │ ├── mepa_ablation_results │ ├── analysis_controller.py │ └── analysis_evaluator.py │ ├── multishot-robnas │ ├── README.md │ ├── run_cfg.sh │ ├── run_final_cfg.sh │ └── test_pred.py │ ├── pim_nas │ ├── example_code_snippet.py │ ├── example_code_snippet_search.yaml │ └── plugin │ │ └── germ_pim_nas.py │ ├── surgery │ ├── README.md │ ├── average_supernets.py │ ├── evaluation.py │ ├── nb101 │ │ ├── README.md │ │ ├── oneshot_cfg.yaml │ │ └── zeroshot_cfg.yaml │ ├── nb201 │ │ ├── README.md │ │ ├── deiso_plateaulr.yaml │ │ ├── run_supernet_training.sh │ │ └── zeroshot.yaml │ ├── nb301 │ │ ├── README.md │ │ ├── config.yaml │ │ └── zeroshot.yaml │ ├── nds │ │ ├── README.md │ │ ├── dump_archyaml_and_gtpkl.py │ │ ├── filter_resnexta_group1.py │ │ ├── plugin │ │ │ └── germ_nds.py │ │ ├── requirements.txt │ │ ├── resnet_ordinal_search.yaml │ │ ├── resnet_search.yaml │ │ ├── resnet_zeroshot.yaml │ │ ├── resnexta_ordinal_search.yaml │ │ ├── resnexta_search.yaml │ │ ├── resnexta_zeroshot.yaml │ │ └── test_germ_nds.py │ ├── run_derive_ckpts.py │ ├── run_oneshot_batch.py │ ├── run_supernet_training.py │ └── run_zero_shot.py │ └── tagates │ ├── README.md │ ├── cfgs │ ├── enas_cfgs │ │ └── tagates.yaml │ ├── nb101_cfgs │ │ ├── tagates.yaml │ │ └── tagates_anytime.yaml │ ├── nb201_cfgs │ │ ├── tagates.yaml │ │ └── tagates_anytime.yaml │ └── nb301_cfgs │ │ ├── tagates.yaml │ │ └── tagates_anytime.yaml │ ├── common_utils.py │ ├── train_enas.py │ ├── train_nasbench101.py │ ├── train_nasbench101_anytime.py │ ├── train_nasbench201.py │ ├── train_nasbench201_anytime.py │ ├── train_nasbench301.py │ └── train_nasbench301_anytime.py ├── pylintrc ├── requirements.cu110.txt ├── requirements.python2.txt ├── requirements.txt ├── scripts ├── convert_cfg_mepa2sw.py ├── dist_train.sh ├── final │ └── model_summary.py ├── generate_germ_search_cfg.py ├── get_data.sh ├── hardware │ ├── latency.py │ └── train_latency_model.py ├── nasbench │ ├── train_cellss_pkl.py │ ├── train_nasbench201_pkl.py │ └── train_nasbench_pkl.py ├── pipeline.py ├── plot_train_curve.py ├── ray_pipeline.py └── requirements.txt ├── setup.cfg ├── setup.py └── tests ├── __init__.py ├── conftest.py ├── test_arch_network.py ├── test_common.py ├── test_controller.py ├── test_dataset.py ├── test_dense.py ├── test_ensemble.py ├── test_evaluator.py ├── test_final_trainer.py ├── test_germ.py ├── test_germ_handlers.py ├── test_hardware.py ├── test_layer2.py ├── test_nasbench.py ├── test_nasbench201.py ├── test_nasbench301.py ├── test_objective.py ├── test_rnn_weights_manager.py ├── test_tune.py ├── test_utils.py ├── test_weights_manager.py └── test_wrapper.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/README.md -------------------------------------------------------------------------------- /aw_nas/VERSION: -------------------------------------------------------------------------------- 1 | 0.4.4-dev1 -------------------------------------------------------------------------------- /aw_nas/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/__init__.py -------------------------------------------------------------------------------- /aw_nas/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/base.py -------------------------------------------------------------------------------- /aw_nas/btcs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/btcs/__init__.py -------------------------------------------------------------------------------- /aw_nas/btcs/enas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/btcs/enas.py -------------------------------------------------------------------------------- /aw_nas/btcs/layer2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/btcs/layer2/__init__.py -------------------------------------------------------------------------------- /aw_nas/btcs/layer2/bi_final_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/btcs/layer2/bi_final_model.py -------------------------------------------------------------------------------- /aw_nas/btcs/layer2/controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/btcs/layer2/controller.py -------------------------------------------------------------------------------- /aw_nas/btcs/layer2/diff_weights_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/btcs/layer2/diff_weights_manager.py -------------------------------------------------------------------------------- /aw_nas/btcs/layer2/final_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/btcs/layer2/final_model.py -------------------------------------------------------------------------------- /aw_nas/btcs/layer2/search_space.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/btcs/layer2/search_space.py -------------------------------------------------------------------------------- /aw_nas/btcs/layer2/weights_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/btcs/layer2/weights_manager.py -------------------------------------------------------------------------------- /aw_nas/btcs/nasbench_101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/btcs/nasbench_101.py -------------------------------------------------------------------------------- /aw_nas/btcs/nasbench_201.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/btcs/nasbench_201.py -------------------------------------------------------------------------------- /aw_nas/btcs/nasbench_201_close.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/btcs/nasbench_201_close.py -------------------------------------------------------------------------------- /aw_nas/btcs/nasbench_301.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/btcs/nasbench_301.py -------------------------------------------------------------------------------- /aw_nas/btcs/zero_cost_proxy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/btcs/zero_cost_proxy.py -------------------------------------------------------------------------------- /aw_nas/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/common.py -------------------------------------------------------------------------------- /aw_nas/controller/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/controller/__init__.py -------------------------------------------------------------------------------- /aw_nas/controller/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/controller/base.py -------------------------------------------------------------------------------- /aw_nas/controller/cars_evo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/controller/cars_evo.py -------------------------------------------------------------------------------- /aw_nas/controller/dense_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/controller/dense_controller.py -------------------------------------------------------------------------------- /aw_nas/controller/differentiable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/controller/differentiable.py -------------------------------------------------------------------------------- /aw_nas/controller/evo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/controller/evo.py -------------------------------------------------------------------------------- /aw_nas/controller/ofa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/controller/ofa.py -------------------------------------------------------------------------------- /aw_nas/controller/predictor_based.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/controller/predictor_based.py -------------------------------------------------------------------------------- /aw_nas/controller/rejection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/controller/rejection.py -------------------------------------------------------------------------------- /aw_nas/controller/rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/controller/rl.py -------------------------------------------------------------------------------- /aw_nas/controller/rl_agents.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/controller/rl_agents.py -------------------------------------------------------------------------------- /aw_nas/controller/rl_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/controller/rl_networks.py -------------------------------------------------------------------------------- /aw_nas/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/dataset/__init__.py -------------------------------------------------------------------------------- /aw_nas/dataset/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/dataset/base.py -------------------------------------------------------------------------------- /aw_nas/dataset/bdd100k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/dataset/bdd100k.py -------------------------------------------------------------------------------- /aw_nas/dataset/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/dataset/cifar10.py -------------------------------------------------------------------------------- /aw_nas/dataset/cifar100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/dataset/cifar100.py -------------------------------------------------------------------------------- /aw_nas/dataset/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/dataset/coco.py -------------------------------------------------------------------------------- /aw_nas/dataset/det_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/dataset/det_augmentation.py -------------------------------------------------------------------------------- /aw_nas/dataset/det_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/dataset/det_transform.py -------------------------------------------------------------------------------- /aw_nas/dataset/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/dataset/imagenet.py -------------------------------------------------------------------------------- /aw_nas/dataset/imagenet_downsample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/dataset/imagenet_downsample.py -------------------------------------------------------------------------------- /aw_nas/dataset/miniimagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/dataset/miniimagenet.py -------------------------------------------------------------------------------- /aw_nas/dataset/ptb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/dataset/ptb.py -------------------------------------------------------------------------------- /aw_nas/dataset/svhn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/dataset/svhn.py -------------------------------------------------------------------------------- /aw_nas/dataset/tiny_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/dataset/tiny_imagenet.py -------------------------------------------------------------------------------- /aw_nas/dataset/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/dataset/voc.py -------------------------------------------------------------------------------- /aw_nas/evaluator/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/evaluator/__init__.py -------------------------------------------------------------------------------- /aw_nas/evaluator/arch_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/evaluator/arch_network.py -------------------------------------------------------------------------------- /aw_nas/evaluator/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/evaluator/base.py -------------------------------------------------------------------------------- /aw_nas/evaluator/bftune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/evaluator/bftune.py -------------------------------------------------------------------------------- /aw_nas/evaluator/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/evaluator/compare.py -------------------------------------------------------------------------------- /aw_nas/evaluator/few_shot_shared_weight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/evaluator/few_shot_shared_weight.py -------------------------------------------------------------------------------- /aw_nas/evaluator/mepa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/evaluator/mepa.py -------------------------------------------------------------------------------- /aw_nas/evaluator/shared_weight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/evaluator/shared_weight.py -------------------------------------------------------------------------------- /aw_nas/evaluator/tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/evaluator/tune.py -------------------------------------------------------------------------------- /aw_nas/final/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/final/__init__.py -------------------------------------------------------------------------------- /aw_nas/final/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/final/base.py -------------------------------------------------------------------------------- /aw_nas/final/bnn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/final/bnn_model.py -------------------------------------------------------------------------------- /aw_nas/final/cnn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/final/cnn_model.py -------------------------------------------------------------------------------- /aw_nas/final/cnn_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/final/cnn_trainer.py -------------------------------------------------------------------------------- /aw_nas/final/dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/final/dense.py -------------------------------------------------------------------------------- /aw_nas/final/det_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/final/det_model.py -------------------------------------------------------------------------------- /aw_nas/final/det_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/final/det_trainer.py -------------------------------------------------------------------------------- /aw_nas/final/general_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/final/general_model.py -------------------------------------------------------------------------------- /aw_nas/final/ofa_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/final/ofa_model.py -------------------------------------------------------------------------------- /aw_nas/final/rnn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/final/rnn_model.py -------------------------------------------------------------------------------- /aw_nas/final/rnn_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/final/rnn_trainer.py -------------------------------------------------------------------------------- /aw_nas/final/wrapper_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/final/wrapper_model.py -------------------------------------------------------------------------------- /aw_nas/final/wrapper_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/final/wrapper_trainer.py -------------------------------------------------------------------------------- /aw_nas/germ/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/germ/README.md -------------------------------------------------------------------------------- /aw_nas/germ/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/germ/__init__.py -------------------------------------------------------------------------------- /aw_nas/germ/decisions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/germ/decisions.py -------------------------------------------------------------------------------- /aw_nas/germ/germ.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/germ/germ.py -------------------------------------------------------------------------------- /aw_nas/germ/mbv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/germ/mbv2.py -------------------------------------------------------------------------------- /aw_nas/germ/mobiledet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/germ/mobiledet.py -------------------------------------------------------------------------------- /aw_nas/germ/nb201.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/germ/nb201.py -------------------------------------------------------------------------------- /aw_nas/germ/necks/__init__.py: -------------------------------------------------------------------------------- 1 | from .fpn import * 2 | -------------------------------------------------------------------------------- /aw_nas/germ/necks/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/germ/necks/fpn.py -------------------------------------------------------------------------------- /aw_nas/germ/searchable_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/germ/searchable_blocks.py -------------------------------------------------------------------------------- /aw_nas/germ/shufflev2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/germ/shufflev2.py -------------------------------------------------------------------------------- /aw_nas/germ/supernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/germ/supernet.py -------------------------------------------------------------------------------- /aw_nas/germ/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/germ/utils.py -------------------------------------------------------------------------------- /aw_nas/hardware/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/hardware/__init__.py -------------------------------------------------------------------------------- /aw_nas/hardware/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/hardware/base.py -------------------------------------------------------------------------------- /aw_nas/hardware/compiler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/hardware/compiler/__init__.py -------------------------------------------------------------------------------- /aw_nas/hardware/compiler/dpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/hardware/compiler/dpu.py -------------------------------------------------------------------------------- /aw_nas/hardware/compiler/xavier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/hardware/compiler/xavier.py -------------------------------------------------------------------------------- /aw_nas/hardware/ofa_obj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/hardware/ofa_obj.py -------------------------------------------------------------------------------- /aw_nas/hardware/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/hardware/utils.py -------------------------------------------------------------------------------- /aw_nas/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/main.py -------------------------------------------------------------------------------- /aw_nas/main_hardware.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/main_hardware.py -------------------------------------------------------------------------------- /aw_nas/objective/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/__init__.py -------------------------------------------------------------------------------- /aw_nas/objective/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/base.py -------------------------------------------------------------------------------- /aw_nas/objective/container.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/container.py -------------------------------------------------------------------------------- /aw_nas/objective/detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/detection.py -------------------------------------------------------------------------------- /aw_nas/objective/detection_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/detection_utils/__init__.py -------------------------------------------------------------------------------- /aw_nas/objective/detection_utils/anchors_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/detection_utils/anchors_generator.py -------------------------------------------------------------------------------- /aw_nas/objective/detection_utils/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/detection_utils/base.py -------------------------------------------------------------------------------- /aw_nas/objective/detection_utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/detection_utils/losses.py -------------------------------------------------------------------------------- /aw_nas/objective/detection_utils/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/detection_utils/matcher.py -------------------------------------------------------------------------------- /aw_nas/objective/detection_utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/detection_utils/metrics.py -------------------------------------------------------------------------------- /aw_nas/objective/detection_utils/post_processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/detection_utils/post_processing.py -------------------------------------------------------------------------------- /aw_nas/objective/fault_injection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/fault_injection.py -------------------------------------------------------------------------------- /aw_nas/objective/flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/flops.py -------------------------------------------------------------------------------- /aw_nas/objective/hardware.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/hardware.py -------------------------------------------------------------------------------- /aw_nas/objective/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/image.py -------------------------------------------------------------------------------- /aw_nas/objective/language.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/language.py -------------------------------------------------------------------------------- /aw_nas/objective/mmdet_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/mmdet_head.py -------------------------------------------------------------------------------- /aw_nas/objective/ofa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/ofa.py -------------------------------------------------------------------------------- /aw_nas/objective/zerocost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/zerocost.py -------------------------------------------------------------------------------- /aw_nas/objective/zeroshot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/objective/zeroshot.py -------------------------------------------------------------------------------- /aw_nas/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/ops/__init__.py -------------------------------------------------------------------------------- /aw_nas/ops/baseline_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/ops/baseline_ops.py -------------------------------------------------------------------------------- /aw_nas/ops/bnn_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/ops/bnn_ops.py -------------------------------------------------------------------------------- /aw_nas/ops/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/ops/ops.py -------------------------------------------------------------------------------- /aw_nas/plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/plugin.py -------------------------------------------------------------------------------- /aw_nas/rollout/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/rollout/__init__.py -------------------------------------------------------------------------------- /aw_nas/rollout/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/rollout/base.py -------------------------------------------------------------------------------- /aw_nas/rollout/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/rollout/compare.py -------------------------------------------------------------------------------- /aw_nas/rollout/dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/rollout/dense.py -------------------------------------------------------------------------------- /aw_nas/rollout/ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/rollout/ensemble.py -------------------------------------------------------------------------------- /aw_nas/rollout/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/rollout/general.py -------------------------------------------------------------------------------- /aw_nas/rollout/mutation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/rollout/mutation.py -------------------------------------------------------------------------------- /aw_nas/rollout/ofa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/rollout/ofa.py -------------------------------------------------------------------------------- /aw_nas/rollout/wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/rollout/wrapper.py -------------------------------------------------------------------------------- /aw_nas/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/trainer/__init__.py -------------------------------------------------------------------------------- /aw_nas/trainer/async_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/trainer/async_trainer.py -------------------------------------------------------------------------------- /aw_nas/trainer/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/trainer/base.py -------------------------------------------------------------------------------- /aw_nas/trainer/meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/trainer/meta.py -------------------------------------------------------------------------------- /aw_nas/trainer/ray_dispatcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/trainer/ray_dispatcher.py -------------------------------------------------------------------------------- /aw_nas/trainer/simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/trainer/simple.py -------------------------------------------------------------------------------- /aw_nas/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/utils/__init__.py -------------------------------------------------------------------------------- /aw_nas/utils/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/utils/box_utils.py -------------------------------------------------------------------------------- /aw_nas/utils/common_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/utils/common_utils.py -------------------------------------------------------------------------------- /aw_nas/utils/elapse_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/utils/elapse_utils.py -------------------------------------------------------------------------------- /aw_nas/utils/exception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/utils/exception.py -------------------------------------------------------------------------------- /aw_nas/utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/utils/log.py -------------------------------------------------------------------------------- /aw_nas/utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/utils/lr_scheduler.py -------------------------------------------------------------------------------- /aw_nas/utils/parallel_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/utils/parallel_utils.py -------------------------------------------------------------------------------- /aw_nas/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/utils/registry.py -------------------------------------------------------------------------------- /aw_nas/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/utils/torch_utils.py -------------------------------------------------------------------------------- /aw_nas/utils/vis_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/utils/vis_utils.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/__init__.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/base.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/close_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/close_net.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/dense.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/diff_super_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/diff_super_net.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/ensemble.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/headers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/headers/__init__.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/headers/anchor_based.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/headers/anchor_based.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/headers/atss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/headers/atss.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/headers/bifpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/headers/bifpn.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/headers/classifiers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/headers/classifiers.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/morphism.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/morphism.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/necks/__init__.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/necks/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/necks/base.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/necks/bifpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/necks/bifpn.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/necks/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/necks/fpn.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/necks/ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/necks/ssd.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/ofa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/ofa.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/ofa_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/ofa_backbone.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/rnn_diff_super_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/rnn_diff_super_net.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/rnn_shared.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/rnn_shared.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/rnn_super_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/rnn_super_net.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/shared.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/shared.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/stub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/stub.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/super_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/super_net.py -------------------------------------------------------------------------------- /aw_nas/weights_manager/wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/aw_nas/weights_manager/wrapper.py -------------------------------------------------------------------------------- /doc/awnas.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/doc/awnas.pdf -------------------------------------------------------------------------------- /doc/darts_flow.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/doc/darts_flow.md -------------------------------------------------------------------------------- /doc/development.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/doc/development.md -------------------------------------------------------------------------------- /doc/enas_cfg_notes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/doc/enas_cfg_notes.md -------------------------------------------------------------------------------- /doc/hardware.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/doc/hardware.md -------------------------------------------------------------------------------- /doc/pics/coverage.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/doc/pics/coverage.svg -------------------------------------------------------------------------------- /doc/pics/enas_cell.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/doc/pics/enas_cell.jpg -------------------------------------------------------------------------------- /doc/pics/flow.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/doc/pics/flow.pdf -------------------------------------------------------------------------------- /doc/pics/framework.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/doc/pics/framework.pdf -------------------------------------------------------------------------------- /doc/pics/hardware_profiling.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/doc/pics/hardware_profiling.pdf -------------------------------------------------------------------------------- /doc/pics/logo_nicsefc.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/doc/pics/logo_nicsefc.jpg -------------------------------------------------------------------------------- /doc/pics/logo_novauto.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/doc/pics/logo_novauto.jpg -------------------------------------------------------------------------------- /examples/baselines/densenet121.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/baselines/densenet121.yaml -------------------------------------------------------------------------------- /examples/baselines/generate_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/baselines/generate_resnet.py -------------------------------------------------------------------------------- /examples/baselines/mobilenet-v2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/baselines/mobilenet-v2.yaml -------------------------------------------------------------------------------- /examples/baselines/resnet18.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/baselines/resnet18.yaml -------------------------------------------------------------------------------- /examples/baselines/resnet18_distributed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/baselines/resnet18_distributed.yaml -------------------------------------------------------------------------------- /examples/baselines/resnet18_imgnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/baselines/resnet18_imgnet.yaml -------------------------------------------------------------------------------- /examples/baselines/resnet20.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/baselines/resnet20.yaml -------------------------------------------------------------------------------- /examples/baselines/resnet50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/baselines/resnet50.yaml -------------------------------------------------------------------------------- /examples/baselines/vgg16.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/baselines/vgg16.yaml -------------------------------------------------------------------------------- /examples/basic/darts.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/basic/darts.yaml -------------------------------------------------------------------------------- /examples/basic/enas.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/basic/enas.yaml -------------------------------------------------------------------------------- /examples/basic/enas_evo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/basic/enas_evo.yaml -------------------------------------------------------------------------------- /examples/basic/enas_multigpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/basic/enas_multigpu.yaml -------------------------------------------------------------------------------- /examples/basic/final_templates/final_template.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/basic/final_templates/final_template.yaml -------------------------------------------------------------------------------- /examples/basic/final_templates/sur_template.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/basic/final_templates/sur_template.yaml -------------------------------------------------------------------------------- /examples/basic/pcdarts.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/basic/pcdarts.yaml -------------------------------------------------------------------------------- /examples/datasets/imagenet/enas_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/datasets/imagenet/enas_search.yaml -------------------------------------------------------------------------------- /examples/datasets/imagenet/final_template.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/datasets/imagenet/final_template.yaml -------------------------------------------------------------------------------- /examples/datasets/imagenet/mepa3_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/datasets/imagenet/mepa3_search.yaml -------------------------------------------------------------------------------- /examples/datasets/imagenet/sur_template.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/datasets/imagenet/sur_template.yaml -------------------------------------------------------------------------------- /examples/datasets/tiny-imagenet/resnet20.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/datasets/tiny-imagenet/resnet20.yaml -------------------------------------------------------------------------------- /examples/detection/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/detection/README.md -------------------------------------------------------------------------------- /examples/detection/VOC_ssd_final_distill.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/detection/VOC_ssd_final_distill.yaml -------------------------------------------------------------------------------- /examples/detection/VOC_ssd_supernet_distill.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/detection/VOC_ssd_supernet_distill.yaml -------------------------------------------------------------------------------- /examples/detection/mmdet_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/detection/mmdet_final.yaml -------------------------------------------------------------------------------- /examples/detection/ssd_async_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/detection/ssd_async_search.yaml -------------------------------------------------------------------------------- /examples/detection/ssd_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/detection/ssd_final.yaml -------------------------------------------------------------------------------- /examples/detection/ssd_supernet_training.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/detection/ssd_supernet_training.yaml -------------------------------------------------------------------------------- /examples/fewshot/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/fewshot/README.md -------------------------------------------------------------------------------- /examples/fewshot/tnas_5way_1shot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/fewshot/tnas_5way_1shot.yaml -------------------------------------------------------------------------------- /examples/fewshot/tnas_5way_1shot_reptile.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/fewshot/tnas_5way_1shot_reptile.yaml -------------------------------------------------------------------------------- /examples/fewshot/tnas_5way_5shot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/fewshot/tnas_5way_5shot.yaml -------------------------------------------------------------------------------- /examples/germ/example_code_snippet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/germ/example_code_snippet.py -------------------------------------------------------------------------------- /examples/germ/example_code_snippet.py.sscfg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/germ/example_code_snippet.py.sscfg.yaml -------------------------------------------------------------------------------- /examples/germ/example_code_snippet_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/germ/example_code_snippet_search.yaml -------------------------------------------------------------------------------- /examples/germ/example_code_snippet_ss.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/germ/example_code_snippet_ss.yaml -------------------------------------------------------------------------------- /examples/germ/mbv2_code_snippet_ss.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/germ/mbv2_code_snippet_ss.yaml -------------------------------------------------------------------------------- /examples/germ/mbv2_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/germ/mbv2_final.yaml -------------------------------------------------------------------------------- /examples/germ/mbv2_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/germ/mbv2_search.yaml -------------------------------------------------------------------------------- /examples/germ/mbv2_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/germ/mbv2_test.yaml -------------------------------------------------------------------------------- /examples/hardware/configs/general.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/hardware/configs/general.yaml -------------------------------------------------------------------------------- /examples/hardware/configs/hardware_obj.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/hardware/configs/hardware_obj.yaml -------------------------------------------------------------------------------- /examples/hardware/configs/ofa_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/hardware/configs/ofa_final.yaml -------------------------------------------------------------------------------- /examples/hardware/configs/ofa_lat.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/hardware/configs/ofa_lat.yaml -------------------------------------------------------------------------------- /examples/hardware/det_ofa_hardware.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/hardware/det_ofa_hardware.yaml -------------------------------------------------------------------------------- /examples/hardware/det_ofa_xavier.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/hardware/det_ofa_xavier.yaml -------------------------------------------------------------------------------- /examples/mloss/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/mloss/README.md -------------------------------------------------------------------------------- /examples/mloss/darts/darts_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/mloss/darts/darts_final.yaml -------------------------------------------------------------------------------- /examples/mloss/darts/darts_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/mloss/darts/darts_search.yaml -------------------------------------------------------------------------------- /examples/mloss/darts/derived_genotypes.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/mloss/darts/derived_genotypes.yaml -------------------------------------------------------------------------------- /examples/mloss/enas/derived_genotypes.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/mloss/enas/derived_genotypes.yaml -------------------------------------------------------------------------------- /examples/mloss/enas/enas_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/mloss/enas/enas_final.yaml -------------------------------------------------------------------------------- /examples/mloss/enas/enas_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/mloss/enas/enas_search.yaml -------------------------------------------------------------------------------- /examples/mloss/pcdarts/pcdarts_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/mloss/pcdarts/pcdarts_final.yaml -------------------------------------------------------------------------------- /examples/mloss/pcdarts/pcdarts_genotypes.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/mloss/pcdarts/pcdarts_genotypes.yaml -------------------------------------------------------------------------------- /examples/mloss/pcdarts/pcdarts_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/mloss/pcdarts/pcdarts_search.yaml -------------------------------------------------------------------------------- /examples/mloss/snas/derived_genotypes.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/mloss/snas/derived_genotypes.yaml -------------------------------------------------------------------------------- /examples/mloss/snas/snas_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/mloss/snas/snas_final.yaml -------------------------------------------------------------------------------- /examples/mloss/snas/snas_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/mloss/snas/snas_search.yaml -------------------------------------------------------------------------------- /examples/nasbench/nasbench-101_1shot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/nasbench/nasbench-101_1shot.yaml -------------------------------------------------------------------------------- /examples/nasbench/nasbench-101_evo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/nasbench/nasbench-101_evo.yaml -------------------------------------------------------------------------------- /examples/nasbench/nasbench-101_gates_sa.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/nasbench/nasbench-101_gates_sa.yaml -------------------------------------------------------------------------------- /examples/nasbench/nasbench-101_sa.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/nasbench/nasbench-101_sa.yaml -------------------------------------------------------------------------------- /examples/nasbench/nasbench-201_final_template.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/nasbench/nasbench-201_final_template.yaml -------------------------------------------------------------------------------- /examples/nasbench/nasbench-201_gates_sa.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/nasbench/nasbench-201_gates_sa.yaml -------------------------------------------------------------------------------- /examples/nasbench/nasbench-301.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/nasbench/nasbench-301.yaml -------------------------------------------------------------------------------- /examples/nasbench/nasbench-301_plateaulr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/nasbench/nasbench-301_plateaulr.yaml -------------------------------------------------------------------------------- /examples/nasbench/nasbench_201_evo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/nasbench/nasbench_201_evo.yaml -------------------------------------------------------------------------------- /examples/nasbench/nasbench_201_sa.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/nasbench/nasbench_201_sa.yaml -------------------------------------------------------------------------------- /examples/ofa/cifar100_mbv3_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/ofa/cifar100_mbv3_search.yaml -------------------------------------------------------------------------------- /examples/ofa/mbv3_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/ofa/mbv3_final.yaml -------------------------------------------------------------------------------- /examples/ofa/mbv3_pareto_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/ofa/mbv3_pareto_search.yaml -------------------------------------------------------------------------------- /examples/ofa/mbv3_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/ofa/mbv3_search.yaml -------------------------------------------------------------------------------- /examples/ofa/mbv3_supernet_training.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/ofa/mbv3_supernet_training.yaml -------------------------------------------------------------------------------- /examples/ofa/mnas_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/ofa/mnas_final.yaml -------------------------------------------------------------------------------- /examples/ofa/mnas_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/ofa/mnas_search.yaml -------------------------------------------------------------------------------- /examples/ofa/mnas_supernet_training.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/ofa/mnas_supernet_training.yaml -------------------------------------------------------------------------------- /examples/plugins/fbnet/fbnet_plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/fbnet/fbnet_plugin.py -------------------------------------------------------------------------------- /examples/plugins/fbnet/sample_fbnet_diff.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/fbnet/sample_fbnet_diff.yaml -------------------------------------------------------------------------------- /examples/plugins/pool/pool_plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/pool/pool_plugin.py -------------------------------------------------------------------------------- /examples/plugins/robustness/1_robustness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/1_robustness.py -------------------------------------------------------------------------------- /examples/plugins/robustness/2_multi_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/2_multi_evaluator.py -------------------------------------------------------------------------------- /examples/plugins/robustness/cfgs/example_supernet_training.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/cfgs/example_supernet_training.yaml -------------------------------------------------------------------------------- /examples/plugins/robustness/cfgs/final/train-cifar10-MSRobNet-1000.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/cfgs/final/train-cifar10-MSRobNet-1000.yaml -------------------------------------------------------------------------------- /examples/plugins/robustness/cfgs/final/train-cifar100-resnet18.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/cfgs/final/train-cifar100-resnet18.yaml -------------------------------------------------------------------------------- /examples/plugins/robustness/cfgs/stagewise_predbased/final_ms2000_predbased.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/cfgs/stagewise_predbased/final_ms2000_predbased.yaml -------------------------------------------------------------------------------- /examples/plugins/robustness/cfgs/stagewise_predbased/search-24c_stagewisess_pgd_gates.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/cfgs/stagewise_predbased/search-24c_stagewisess_pgd_gates.yaml -------------------------------------------------------------------------------- /examples/plugins/robustness/cfgs/stagewise_predbased/search-multishoteval_stagewisess_fgsm_gates_2000M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/cfgs/stagewise_predbased/search-multishoteval_stagewisess_fgsm_gates_2000M.yaml -------------------------------------------------------------------------------- /examples/plugins/robustness/cfgs/stagewise_predbased/search-multishoteval_stagewisess_pgd_gates_2000M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/cfgs/stagewise_predbased/search-multishoteval_stagewisess_pgd_gates_2000M.yaml -------------------------------------------------------------------------------- /examples/plugins/robustness/robustness_archembedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/robustness_archembedder.py -------------------------------------------------------------------------------- /examples/plugins/robustness/robustness_final_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/robustness_final_model.py -------------------------------------------------------------------------------- /examples/plugins/robustness/robustness_ss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/robustness_ss.py -------------------------------------------------------------------------------- /examples/plugins/robustness/robustness_weights_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/robustness_weights_manager.py -------------------------------------------------------------------------------- /examples/plugins/robustness/test_acc_attack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/test_acc_attack.py -------------------------------------------------------------------------------- /examples/plugins/robustness/test_black_attack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/test_black_attack.py -------------------------------------------------------------------------------- /examples/plugins/robustness/test_distance_attack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/test_distance_attack.py -------------------------------------------------------------------------------- /examples/plugins/robustness/test_rob_ss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/plugins/robustness/test_rob_ss.py -------------------------------------------------------------------------------- /examples/research/bbssp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/bbssp/README.md -------------------------------------------------------------------------------- /examples/research/bbssp/diff_config/diff_cifar10_allblock.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/bbssp/diff_config/diff_cifar10_allblock.yaml -------------------------------------------------------------------------------- /examples/research/bbssp/diff_config/speed_diff_allblock.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/bbssp/diff_config/speed_diff_allblock.txt -------------------------------------------------------------------------------- /examples/research/bbssp/rl_config/rl_cifar10_allblock.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/bbssp/rl_config/rl_cifar10_allblock.yaml -------------------------------------------------------------------------------- /examples/research/bbssp/rl_config/speed_cifar_allblock.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/bbssp/rl_config/speed_cifar_allblock.txt -------------------------------------------------------------------------------- /examples/research/bbssp/ssp_plugin/ssp_plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/bbssp/ssp_plugin/ssp_plugin.py -------------------------------------------------------------------------------- /examples/research/bnn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/bnn/README.md -------------------------------------------------------------------------------- /examples/research/close/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/close/README.md -------------------------------------------------------------------------------- /examples/research/close/nb201/node_cl-2468_lr-4_mlp-200x3_wit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/close/nb201/node_cl-2468_lr-4_mlp-200x3_wit.yaml -------------------------------------------------------------------------------- /examples/research/close/nb201/node_cl-246_lr-4_mlp-200x3_wit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/close/nb201/node_cl-246_lr-4_mlp-200x3_wit.yaml -------------------------------------------------------------------------------- /examples/research/close/nb201/node_cl-24_lr-4_mlp-200x3_wit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/close/nb201/node_cl-24_lr-4_mlp-200x3_wit.yaml -------------------------------------------------------------------------------- /examples/research/close/nb301/node_cl-2468_lr-4_mlp-200x3_wit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/close/nb301/node_cl-2468_lr-4_mlp-200x3_wit.yaml -------------------------------------------------------------------------------- /examples/research/close/nb301/node_cl-246_lr-4_mlp-200x3_wit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/close/nb301/node_cl-246_lr-4_mlp-200x3_wit.yaml -------------------------------------------------------------------------------- /examples/research/close/nb301/node_cl-24_lr-4_mlp-200x3_wit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/close/nb301/node_cl-24_lr-4_mlp-200x3_wit.yaml -------------------------------------------------------------------------------- /examples/research/close/nds/resnet_cl-9-22446688_lr-4_pre.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/close/nds/resnet_cl-9-22446688_lr-4_pre.yaml -------------------------------------------------------------------------------- /examples/research/close/nds/resnexta_cl-9-2468_lr-4_pre.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/close/nds/resnexta_cl-9-2468_lr-4_pre.yaml -------------------------------------------------------------------------------- /examples/research/close/run_derive_ckpts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/close/run_derive_ckpts.py -------------------------------------------------------------------------------- /examples/research/close/run_derive_epoch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/close/run_derive_epoch.py -------------------------------------------------------------------------------- /examples/research/close/run_supernet_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/close/run_supernet_training.py -------------------------------------------------------------------------------- /examples/research/dele/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/dele/README.md -------------------------------------------------------------------------------- /examples/research/dele/cfgs/dynamic_ensemble_nb201_gates_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/dele/cfgs/dynamic_ensemble_nb201_gates_config.yaml -------------------------------------------------------------------------------- /examples/research/dele/cfgs/nasbench-201_gates_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/dele/cfgs/nasbench-201_gates_search.yaml -------------------------------------------------------------------------------- /examples/research/dele/cfgs/train_nb201_gates_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/dele/cfgs/train_nb201_gates_config.yaml -------------------------------------------------------------------------------- /examples/research/dele/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/dele/dataset.py -------------------------------------------------------------------------------- /examples/research/dele/dynamic_ensemble_train_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/dele/dynamic_ensemble_train_predictor.py -------------------------------------------------------------------------------- /examples/research/dele/plugin/nds_arch_embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/dele/plugin/nds_arch_embedder.py -------------------------------------------------------------------------------- /examples/research/dele/train_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/dele/train_predictor.py -------------------------------------------------------------------------------- /examples/research/dele/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/dele/utils.py -------------------------------------------------------------------------------- /examples/research/ftt-nas/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/ftt-nas/README.md -------------------------------------------------------------------------------- /examples/research/ftt-nas/adsaf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/ftt-nas/adsaf.yaml -------------------------------------------------------------------------------- /examples/research/ftt-nas/adsaf_final.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/ftt-nas/adsaf_final.yaml -------------------------------------------------------------------------------- /examples/research/ftt-nas/fixed_point_plugins/fixed_point_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/ftt-nas/fixed_point_plugins/fixed_point_patch.py -------------------------------------------------------------------------------- /examples/research/ftt-nas/fixed_point_plugins/fixed_point_patch_new.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/ftt-nas/fixed_point_plugins/fixed_point_patch_new.py -------------------------------------------------------------------------------- /examples/research/ftt-nas/fixed_point_plugins/fixed_point_rram_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/ftt-nas/fixed_point_plugins/fixed_point_rram_patch.py -------------------------------------------------------------------------------- /examples/research/ftt-nas/fixed_point_plugins/fixed_point_rram_patch_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/ftt-nas/fixed_point_plugins/fixed_point_rram_patch_all.py -------------------------------------------------------------------------------- /examples/research/ftt-nas/fixed_point_plugins/fixed_point_rram_patch_bit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/ftt-nas/fixed_point_plugins/fixed_point_rram_patch_bit.py -------------------------------------------------------------------------------- /examples/research/ftt-nas/mibb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/ftt-nas/mibb.yaml -------------------------------------------------------------------------------- /examples/research/ftt-nas/run_adsaf.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/ftt-nas/run_adsaf.sh -------------------------------------------------------------------------------- /examples/research/ftt-nas/run_adsaf_search.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/ftt-nas/run_adsaf_search.sh -------------------------------------------------------------------------------- /examples/research/ftt-nas/run_mibb.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/ftt-nas/run_mibb.sh -------------------------------------------------------------------------------- /examples/research/ftt-nas/run_mibb_search.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/ftt-nas/run_mibb_search.sh -------------------------------------------------------------------------------- /examples/research/ftt-nas/test_fix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/ftt-nas/test_fix.py -------------------------------------------------------------------------------- /examples/research/gates/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/gates/README.md -------------------------------------------------------------------------------- /examples/research/gates/dump_nb101_allv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/gates/dump_nb101_allv.py -------------------------------------------------------------------------------- /examples/research/gates/nbc_flow_gcn128x4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/gates/nbc_flow_gcn128x4.yaml -------------------------------------------------------------------------------- /examples/research/gates/nbc_flow_gcn128x4_pairdiff.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/gates/nbc_flow_gcn128x4_pairdiff.yaml -------------------------------------------------------------------------------- /examples/research/gates/print_res.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/gates/print_res.sh -------------------------------------------------------------------------------- /examples/research/gibbon/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/gibbon/README.md -------------------------------------------------------------------------------- /examples/research/mepa_ablation_results/analysis_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/mepa_ablation_results/analysis_controller.py -------------------------------------------------------------------------------- /examples/research/mepa_ablation_results/analysis_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/mepa_ablation_results/analysis_evaluator.py -------------------------------------------------------------------------------- /examples/research/multishot-robnas/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/multishot-robnas/README.md -------------------------------------------------------------------------------- /examples/research/multishot-robnas/run_cfg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/multishot-robnas/run_cfg.sh -------------------------------------------------------------------------------- /examples/research/multishot-robnas/run_final_cfg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/multishot-robnas/run_final_cfg.sh -------------------------------------------------------------------------------- /examples/research/multishot-robnas/test_pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/multishot-robnas/test_pred.py -------------------------------------------------------------------------------- /examples/research/pim_nas/example_code_snippet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/pim_nas/example_code_snippet.py -------------------------------------------------------------------------------- /examples/research/pim_nas/example_code_snippet_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/pim_nas/example_code_snippet_search.yaml -------------------------------------------------------------------------------- /examples/research/pim_nas/plugin/germ_pim_nas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/pim_nas/plugin/germ_pim_nas.py -------------------------------------------------------------------------------- /examples/research/surgery/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/README.md -------------------------------------------------------------------------------- /examples/research/surgery/average_supernets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/average_supernets.py -------------------------------------------------------------------------------- /examples/research/surgery/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/evaluation.py -------------------------------------------------------------------------------- /examples/research/surgery/nb101/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nb101/README.md -------------------------------------------------------------------------------- /examples/research/surgery/nb101/oneshot_cfg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nb101/oneshot_cfg.yaml -------------------------------------------------------------------------------- /examples/research/surgery/nb101/zeroshot_cfg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nb101/zeroshot_cfg.yaml -------------------------------------------------------------------------------- /examples/research/surgery/nb201/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nb201/README.md -------------------------------------------------------------------------------- /examples/research/surgery/nb201/deiso_plateaulr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nb201/deiso_plateaulr.yaml -------------------------------------------------------------------------------- /examples/research/surgery/nb201/run_supernet_training.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nb201/run_supernet_training.sh -------------------------------------------------------------------------------- /examples/research/surgery/nb201/zeroshot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nb201/zeroshot.yaml -------------------------------------------------------------------------------- /examples/research/surgery/nb301/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nb301/README.md -------------------------------------------------------------------------------- /examples/research/surgery/nb301/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nb301/config.yaml -------------------------------------------------------------------------------- /examples/research/surgery/nb301/zeroshot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nb301/zeroshot.yaml -------------------------------------------------------------------------------- /examples/research/surgery/nds/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nds/README.md -------------------------------------------------------------------------------- /examples/research/surgery/nds/dump_archyaml_and_gtpkl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nds/dump_archyaml_and_gtpkl.py -------------------------------------------------------------------------------- /examples/research/surgery/nds/filter_resnexta_group1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nds/filter_resnexta_group1.py -------------------------------------------------------------------------------- /examples/research/surgery/nds/plugin/germ_nds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nds/plugin/germ_nds.py -------------------------------------------------------------------------------- /examples/research/surgery/nds/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nds/requirements.txt -------------------------------------------------------------------------------- /examples/research/surgery/nds/resnet_ordinal_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nds/resnet_ordinal_search.yaml -------------------------------------------------------------------------------- /examples/research/surgery/nds/resnet_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nds/resnet_search.yaml -------------------------------------------------------------------------------- /examples/research/surgery/nds/resnet_zeroshot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nds/resnet_zeroshot.yaml -------------------------------------------------------------------------------- /examples/research/surgery/nds/resnexta_ordinal_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nds/resnexta_ordinal_search.yaml -------------------------------------------------------------------------------- /examples/research/surgery/nds/resnexta_search.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nds/resnexta_search.yaml -------------------------------------------------------------------------------- /examples/research/surgery/nds/resnexta_zeroshot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nds/resnexta_zeroshot.yaml -------------------------------------------------------------------------------- /examples/research/surgery/nds/test_germ_nds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/nds/test_germ_nds.py -------------------------------------------------------------------------------- /examples/research/surgery/run_derive_ckpts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/run_derive_ckpts.py -------------------------------------------------------------------------------- /examples/research/surgery/run_oneshot_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/run_oneshot_batch.py -------------------------------------------------------------------------------- /examples/research/surgery/run_supernet_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/run_supernet_training.py -------------------------------------------------------------------------------- /examples/research/surgery/run_zero_shot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/surgery/run_zero_shot.py -------------------------------------------------------------------------------- /examples/research/tagates/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/tagates/README.md -------------------------------------------------------------------------------- /examples/research/tagates/cfgs/enas_cfgs/tagates.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/tagates/cfgs/enas_cfgs/tagates.yaml -------------------------------------------------------------------------------- /examples/research/tagates/cfgs/nb101_cfgs/tagates.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/tagates/cfgs/nb101_cfgs/tagates.yaml -------------------------------------------------------------------------------- /examples/research/tagates/cfgs/nb101_cfgs/tagates_anytime.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/tagates/cfgs/nb101_cfgs/tagates_anytime.yaml -------------------------------------------------------------------------------- /examples/research/tagates/cfgs/nb201_cfgs/tagates.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/tagates/cfgs/nb201_cfgs/tagates.yaml -------------------------------------------------------------------------------- /examples/research/tagates/cfgs/nb201_cfgs/tagates_anytime.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/tagates/cfgs/nb201_cfgs/tagates_anytime.yaml -------------------------------------------------------------------------------- /examples/research/tagates/cfgs/nb301_cfgs/tagates.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/tagates/cfgs/nb301_cfgs/tagates.yaml -------------------------------------------------------------------------------- /examples/research/tagates/cfgs/nb301_cfgs/tagates_anytime.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/tagates/cfgs/nb301_cfgs/tagates_anytime.yaml -------------------------------------------------------------------------------- /examples/research/tagates/common_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/tagates/common_utils.py -------------------------------------------------------------------------------- /examples/research/tagates/train_enas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/tagates/train_enas.py -------------------------------------------------------------------------------- /examples/research/tagates/train_nasbench101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/tagates/train_nasbench101.py -------------------------------------------------------------------------------- /examples/research/tagates/train_nasbench101_anytime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/tagates/train_nasbench101_anytime.py -------------------------------------------------------------------------------- /examples/research/tagates/train_nasbench201.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/tagates/train_nasbench201.py -------------------------------------------------------------------------------- /examples/research/tagates/train_nasbench201_anytime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/tagates/train_nasbench201_anytime.py -------------------------------------------------------------------------------- /examples/research/tagates/train_nasbench301.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/tagates/train_nasbench301.py -------------------------------------------------------------------------------- /examples/research/tagates/train_nasbench301_anytime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/examples/research/tagates/train_nasbench301_anytime.py -------------------------------------------------------------------------------- /pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/pylintrc -------------------------------------------------------------------------------- /requirements.cu110.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/requirements.cu110.txt -------------------------------------------------------------------------------- /requirements.python2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/requirements.python2.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/convert_cfg_mepa2sw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/scripts/convert_cfg_mepa2sw.py -------------------------------------------------------------------------------- /scripts/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/scripts/dist_train.sh -------------------------------------------------------------------------------- /scripts/final/model_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/scripts/final/model_summary.py -------------------------------------------------------------------------------- /scripts/generate_germ_search_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/scripts/generate_germ_search_cfg.py -------------------------------------------------------------------------------- /scripts/get_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/scripts/get_data.sh -------------------------------------------------------------------------------- /scripts/hardware/latency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/scripts/hardware/latency.py -------------------------------------------------------------------------------- /scripts/hardware/train_latency_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/scripts/hardware/train_latency_model.py -------------------------------------------------------------------------------- /scripts/nasbench/train_cellss_pkl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/scripts/nasbench/train_cellss_pkl.py -------------------------------------------------------------------------------- /scripts/nasbench/train_nasbench201_pkl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/scripts/nasbench/train_nasbench201_pkl.py -------------------------------------------------------------------------------- /scripts/nasbench/train_nasbench_pkl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/scripts/nasbench/train_nasbench_pkl.py -------------------------------------------------------------------------------- /scripts/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/scripts/pipeline.py -------------------------------------------------------------------------------- /scripts/plot_train_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/scripts/plot_train_curve.py -------------------------------------------------------------------------------- /scripts/ray_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/scripts/ray_pipeline.py -------------------------------------------------------------------------------- /scripts/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/scripts/requirements.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- 1 | [wheel] 2 | universal = 1 3 | 4 | 5 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/test_arch_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_arch_network.py -------------------------------------------------------------------------------- /tests/test_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_common.py -------------------------------------------------------------------------------- /tests/test_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_controller.py -------------------------------------------------------------------------------- /tests/test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_dataset.py -------------------------------------------------------------------------------- /tests/test_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_dense.py -------------------------------------------------------------------------------- /tests/test_ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_ensemble.py -------------------------------------------------------------------------------- /tests/test_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_evaluator.py -------------------------------------------------------------------------------- /tests/test_final_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_final_trainer.py -------------------------------------------------------------------------------- /tests/test_germ.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_germ.py -------------------------------------------------------------------------------- /tests/test_germ_handlers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_germ_handlers.py -------------------------------------------------------------------------------- /tests/test_hardware.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_hardware.py -------------------------------------------------------------------------------- /tests/test_layer2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_layer2.py -------------------------------------------------------------------------------- /tests/test_nasbench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_nasbench.py -------------------------------------------------------------------------------- /tests/test_nasbench201.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_nasbench201.py -------------------------------------------------------------------------------- /tests/test_nasbench301.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_nasbench301.py -------------------------------------------------------------------------------- /tests/test_objective.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_objective.py -------------------------------------------------------------------------------- /tests/test_rnn_weights_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_rnn_weights_manager.py -------------------------------------------------------------------------------- /tests/test_tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_tune.py -------------------------------------------------------------------------------- /tests/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_utils.py -------------------------------------------------------------------------------- /tests/test_weights_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_weights_manager.py -------------------------------------------------------------------------------- /tests/test_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/walkerning/aw_nas/HEAD/tests/test_wrapper.py --------------------------------------------------------------------------------