Merging-MobileNets-for-Multitask
Official implementation of On Merging MobileNets for Efficient Multitask Inference.
Created by Cheng-En Wu , Yi-Ming Chan(yiming@iis.sinica.edu.tw), Chu-Song Chen(song@iis.sinica.edu.tw)
The code is released for academic research use only. For commercial use, please contact Dr. Chu-Song Chen(song@iis.sinica.edu.tw).
Usage
Pretrained and merged checkpoints are available here: https://reurl.cc/96o3x
Datasets in TFRecords are available here: https://reurl.cc/vp3vN
All rights belong to the respective publishers. The datasets are provided only to aid reproducibility.
Place and unzip pretrained checkpoints checkpoints.tar.gz
in checkpoints/ ,
datasets.tar.gz
in datasets/ hungarian_algorithm.tar.gz
in hungarian_algorithm/
Mergeing
Check out convert_ckpt_to_npy.sh
and merge_layers.sh
Convert pretrained checkpoints to numpy files and merge pretrained checkpoints into a unified checkpoints.
Training
Check out zipper_multiple_train.sh
Excute training in zippering process.
Inference
Check out zipper_eval_script.sh
Evaluate the Top-1 accuracy.
Citation
Please cite following paper if these codes help your research:
@inproceedings{wu2019mobilenet_merging,
Title = {On Merging MobileNets for Efficient Multitask Inference},
Author = {Cheng-En Wu, Yi-Ming Chan and Chu-Song Chen},
booktitle = {2019 IEEE International Symposium on High-Performance Computer Architecture Workshop},
year = {2019}
}
Contact
Please feel free to leave suggestions or comments to Cheng-En Wu(chengen@iis.sinica.edu.tw) , Yi-Ming Chan(yiming@iis.sinica.edu.tw), Chu-Song Chen(song@iis.sinica.edu.tw)