├── .gitignore ├── README.md ├── data_gt └── multi30k │ ├── gt_de.txt │ └── gt_en.txt ├── datasets └── imagenet_dataset.py ├── evaluate ├── imagenet_validataion.py ├── translate.py └── translator.py ├── models ├── students │ ├── resnet18.py │ ├── resnet34.py │ ├── resnet50.py │ └── transformer.py └── teachers │ ├── .gitignore │ ├── resnet.py │ └── transformer.py ├── operators ├── dorefa │ ├── data_type.py │ ├── dorefa.py │ └── quantize_helper.py └── lsq │ ├── lsq.py │ └── lsq_modules.py ├── requirements.txt ├── train_imagenet ├── train_archnet_resnet18.py ├── train_archnet_resnet34.py └── train_archnet_resnet50.py └── train_transformer └── train_archnet_transformer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/README.md -------------------------------------------------------------------------------- /data_gt/multi30k/gt_de.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/data_gt/multi30k/gt_de.txt -------------------------------------------------------------------------------- /data_gt/multi30k/gt_en.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/data_gt/multi30k/gt_en.txt -------------------------------------------------------------------------------- /datasets/imagenet_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/datasets/imagenet_dataset.py -------------------------------------------------------------------------------- /evaluate/imagenet_validataion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/evaluate/imagenet_validataion.py -------------------------------------------------------------------------------- /evaluate/translate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/evaluate/translate.py -------------------------------------------------------------------------------- /evaluate/translator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/evaluate/translator.py -------------------------------------------------------------------------------- /models/students/resnet18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/models/students/resnet18.py -------------------------------------------------------------------------------- /models/students/resnet34.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/models/students/resnet34.py -------------------------------------------------------------------------------- /models/students/resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/models/students/resnet50.py -------------------------------------------------------------------------------- /models/students/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/models/students/transformer.py -------------------------------------------------------------------------------- /models/teachers/.gitignore: -------------------------------------------------------------------------------- 1 | pretrained_models 2 | *.pth 3 | *.tar 4 | -------------------------------------------------------------------------------- /models/teachers/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/models/teachers/resnet.py -------------------------------------------------------------------------------- /models/teachers/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/models/teachers/transformer.py -------------------------------------------------------------------------------- /operators/dorefa/data_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/operators/dorefa/data_type.py -------------------------------------------------------------------------------- /operators/dorefa/dorefa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/operators/dorefa/dorefa.py -------------------------------------------------------------------------------- /operators/dorefa/quantize_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/operators/dorefa/quantize_helper.py -------------------------------------------------------------------------------- /operators/lsq/lsq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/operators/lsq/lsq.py -------------------------------------------------------------------------------- /operators/lsq/lsq_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/operators/lsq/lsq_modules.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/requirements.txt -------------------------------------------------------------------------------- /train_imagenet/train_archnet_resnet18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/train_imagenet/train_archnet_resnet18.py -------------------------------------------------------------------------------- /train_imagenet/train_archnet_resnet34.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/train_imagenet/train_archnet_resnet34.py -------------------------------------------------------------------------------- /train_imagenet/train_archnet_resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/train_imagenet/train_archnet_resnet50.py -------------------------------------------------------------------------------- /train_transformer/train_archnet_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Arch-Net/HEAD/train_transformer/train_archnet_transformer.py --------------------------------------------------------------------------------