算法部署 | 在内存有限的微控制器上部署ViT算法

• 项目应用场景

• 面向内存有限的微控制器上部署视觉 Transformer 场景，属于低算力平台、边缘计算场景的算法部署项目。

• 项目细节 ==> 具体参见项目 README.md

• (1) 安装依赖

conda create -n MCUFormer python=3.7
conda activate MCUFormer
pip install -r requirements.txt

• (2) 执行分发

python -m torch.distributed.launch --nproc_per_node=8 --use_env supernet_train.py --data-path /PATH/TO/IMAGENT --gp \
--change_qk --relative_position --mode super --dist-eval --load-pretrained-model \
--cfg ./experiments/supernet/supernet-T.yaml --cfg-new  ./experiments/supernet/supernet-T-new.yaml \
--epochs 500 --warmup-epochs 5 --lr 1e-4 --super_epoch 1 --step-num 7 \
--model deit_tiny_patch16_224 --batch-size 128 --output /OUTPUT_PATH

• (3) 搜索优化

python ./lib/subImageNet.py --data-path /PATH/TO/IMAGENT