├── .gitignore ├── .style.yapf ├── LICENSE ├── README.md ├── apps ├── eval │ ├── eval.yml │ ├── eval_se.yml │ └── eval_shrink.yml ├── mobilenet │ ├── default.yml │ ├── default_mnas_scheduler.yml │ ├── mobilenet_v2_mnas.yml │ ├── models │ │ ├── fix_first │ │ │ ├── supernet_single_path_nas.yml │ │ │ └── supernet_single_path_nas_relu.yml │ │ ├── mobilenet_v2_1.0.yml │ │ ├── mobilenet_v2_1.0_relu.yml │ │ ├── proxyless_mobile.yml │ │ └── single_path_nas_final.yml │ ├── proxyless_mobile_mnas.yml │ └── single_path_nas_mnas.yml └── searched │ ├── atomnas_a │ ├── atomnas_a+.yml │ └── atomnas_a.yml │ ├── atomnas_b │ ├── atomnas_b+.yml │ └── atomnas_b.yml │ ├── atomnas_c │ ├── atomnas_c+.yml │ └── atomnas_c.yml │ ├── autonl │ ├── autonl_l.yml │ └── autonl_s.yml │ └── models │ ├── atomnas_a.yml │ ├── atomnas_b.yml │ ├── atomnas_c.yml │ └── autonl │ ├── autonl_l.yml │ └── autonl_s.yml ├── common.py ├── models ├── __init__.py ├── mobilenet_base.py ├── mobilenet_supernet.py └── searched_network.py ├── requirements.txt ├── scripts ├── run.sh ├── run_non_distributed_copy.sh ├── run_non_distributed_no_copy.sh └── utils │ ├── export_data.sh │ ├── remote_run.sh │ ├── setup.sh │ ├── setup_network.sh │ └── utils.sh ├── tests ├── models │ └── mobilenet_base_test.py ├── op │ └── tf_torch_conv_test.py └── utils │ └── optim_test.py ├── tool ├── dump_tf_checkpoint.py ├── modify_searched.py └── parse_tf_dump.py ├── train.py ├── utils ├── __init__.py ├── common.py ├── config.py ├── dataflow.py ├── distributed.py ├── lmdb_dataset.py ├── meters.py ├── model_profiling.py ├── optim.py ├── rmsprop.py ├── test.py └── transforms.py └── val.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/.gitignore -------------------------------------------------------------------------------- /.style.yapf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/.style.yapf -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/README.md -------------------------------------------------------------------------------- /apps/eval/eval.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/eval/eval.yml -------------------------------------------------------------------------------- /apps/eval/eval_se.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/eval/eval_se.yml -------------------------------------------------------------------------------- /apps/eval/eval_shrink.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/eval/eval_shrink.yml -------------------------------------------------------------------------------- /apps/mobilenet/default.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/mobilenet/default.yml -------------------------------------------------------------------------------- /apps/mobilenet/default_mnas_scheduler.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/mobilenet/default_mnas_scheduler.yml -------------------------------------------------------------------------------- /apps/mobilenet/mobilenet_v2_mnas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/mobilenet/mobilenet_v2_mnas.yml -------------------------------------------------------------------------------- /apps/mobilenet/models/fix_first/supernet_single_path_nas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/mobilenet/models/fix_first/supernet_single_path_nas.yml -------------------------------------------------------------------------------- /apps/mobilenet/models/fix_first/supernet_single_path_nas_relu.yml: -------------------------------------------------------------------------------- 1 | _default: !include ./supernet_single_path_nas.yml 2 | 3 | 'active_fn': 'nn.ReLU' 4 | -------------------------------------------------------------------------------- /apps/mobilenet/models/mobilenet_v2_1.0.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/mobilenet/models/mobilenet_v2_1.0.yml -------------------------------------------------------------------------------- /apps/mobilenet/models/mobilenet_v2_1.0_relu.yml: -------------------------------------------------------------------------------- 1 | _default: !include ./mobilenet_v2_1.0.yml 2 | 3 | 'active_fn': 'nn.ReLU' 4 | -------------------------------------------------------------------------------- /apps/mobilenet/models/proxyless_mobile.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/mobilenet/models/proxyless_mobile.yml -------------------------------------------------------------------------------- /apps/mobilenet/models/single_path_nas_final.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/mobilenet/models/single_path_nas_final.yml -------------------------------------------------------------------------------- /apps/mobilenet/proxyless_mobile_mnas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/mobilenet/proxyless_mobile_mnas.yml -------------------------------------------------------------------------------- /apps/mobilenet/single_path_nas_mnas.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/mobilenet/single_path_nas_mnas.yml -------------------------------------------------------------------------------- /apps/searched/atomnas_a/atomnas_a+.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/searched/atomnas_a/atomnas_a+.yml -------------------------------------------------------------------------------- /apps/searched/atomnas_a/atomnas_a.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/searched/atomnas_a/atomnas_a.yml -------------------------------------------------------------------------------- /apps/searched/atomnas_b/atomnas_b+.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/searched/atomnas_b/atomnas_b+.yml -------------------------------------------------------------------------------- /apps/searched/atomnas_b/atomnas_b.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/searched/atomnas_b/atomnas_b.yml -------------------------------------------------------------------------------- /apps/searched/atomnas_c/atomnas_c+.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/searched/atomnas_c/atomnas_c+.yml -------------------------------------------------------------------------------- /apps/searched/atomnas_c/atomnas_c.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/searched/atomnas_c/atomnas_c.yml -------------------------------------------------------------------------------- /apps/searched/autonl/autonl_l.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/searched/autonl/autonl_l.yml -------------------------------------------------------------------------------- /apps/searched/autonl/autonl_s.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/searched/autonl/autonl_s.yml -------------------------------------------------------------------------------- /apps/searched/models/atomnas_a.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/searched/models/atomnas_a.yml -------------------------------------------------------------------------------- /apps/searched/models/atomnas_b.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/searched/models/atomnas_b.yml -------------------------------------------------------------------------------- /apps/searched/models/atomnas_c.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/searched/models/atomnas_c.yml -------------------------------------------------------------------------------- /apps/searched/models/autonl/autonl_l.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/searched/models/autonl/autonl_l.yml -------------------------------------------------------------------------------- /apps/searched/models/autonl/autonl_s.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/apps/searched/models/autonl/autonl_s.yml -------------------------------------------------------------------------------- /common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/common.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/mobilenet_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/models/mobilenet_base.py -------------------------------------------------------------------------------- /models/mobilenet_supernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/models/mobilenet_supernet.py -------------------------------------------------------------------------------- /models/searched_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/models/searched_network.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/scripts/run.sh -------------------------------------------------------------------------------- /scripts/run_non_distributed_copy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/scripts/run_non_distributed_copy.sh -------------------------------------------------------------------------------- /scripts/run_non_distributed_no_copy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/scripts/run_non_distributed_no_copy.sh -------------------------------------------------------------------------------- /scripts/utils/export_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/scripts/utils/export_data.sh -------------------------------------------------------------------------------- /scripts/utils/remote_run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/scripts/utils/remote_run.sh -------------------------------------------------------------------------------- /scripts/utils/setup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/scripts/utils/setup.sh -------------------------------------------------------------------------------- /scripts/utils/setup_network.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/scripts/utils/setup_network.sh -------------------------------------------------------------------------------- /scripts/utils/utils.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/scripts/utils/utils.sh -------------------------------------------------------------------------------- /tests/models/mobilenet_base_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/tests/models/mobilenet_base_test.py -------------------------------------------------------------------------------- /tests/op/tf_torch_conv_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/tests/op/tf_torch_conv_test.py -------------------------------------------------------------------------------- /tests/utils/optim_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/tests/utils/optim_test.py -------------------------------------------------------------------------------- /tool/dump_tf_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/tool/dump_tf_checkpoint.py -------------------------------------------------------------------------------- /tool/modify_searched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/tool/modify_searched.py -------------------------------------------------------------------------------- /tool/parse_tf_dump.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/tool/parse_tf_dump.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/utils/common.py -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/dataflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/utils/dataflow.py -------------------------------------------------------------------------------- /utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/utils/distributed.py -------------------------------------------------------------------------------- /utils/lmdb_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/utils/lmdb_dataset.py -------------------------------------------------------------------------------- /utils/meters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/utils/meters.py -------------------------------------------------------------------------------- /utils/model_profiling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/utils/model_profiling.py -------------------------------------------------------------------------------- /utils/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/utils/optim.py -------------------------------------------------------------------------------- /utils/rmsprop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/utils/rmsprop.py -------------------------------------------------------------------------------- /utils/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/utils/test.py -------------------------------------------------------------------------------- /utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/utils/transforms.py -------------------------------------------------------------------------------- /val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/meijieru/yet_another_mobilenet_series/HEAD/val.py --------------------------------------------------------------------------------