如算法概述表所示,GADS算法基于Python3.6+环境下的numpy、pytorch及tensorflow等依赖库开发。算法输入为由深度模型及部署情境,构建的分割状态图G,输出为算法根据动态运行情境快速自适应调优后的模型最优分割点。该算法支持在CiFar-10、CiFar-100、ImageNet、BDD100K等数据集上的深度卷积网络最优分割点的调优。在代码的实现过程中,GADS算法参考了Neurosurgeon、DADS等第三方资源实现。
算法名称 | 基于图的深度模型自适应手术刀算法 (Graph-based
Adaptive DNN Surgery, GADS) |
算法接口 | python gads.py --config_file config.yaml |
输入 | 由深度模型及部署情境,构建的分割状态图G |
输出 | 根据动态运行情境快速自适应调优的模型分割点 |
gitlab链接:GADS · GitLab (crowdhmt.com)
如果觉得我的文章对您有用,请随意打赏。你的支持将鼓励我继续创作!