- # For 完成模型的边缘部署,实现侧载模型的要求
- # -------------------------------------------------------------------------------------------------------------------- #
- python export_onnx.py --weight './checkpoints/Alexnet/clean/prune_best.pt' --input_size 32 --save_path './checkpoints/Alexnet/clean'
|