1 轻量化模型 一般而言,模型轻量化有三个途径: 知识蒸馏:大模型引导小模型训练,让其逼近大模型效果 轻量化模块替换:利用一些轻量化模块进行替换,减少模型参数 剪枝:通过优化算法引导模型裁剪无用的参数 MobileNetV3论文如下,自行搜索 2 修改步骤 在nn/modules的文件夹下新建一个backbone文件夹,在backbone文件夹里面新建一个MobileNetV3.py文件,放入MobileNetV3源码。 MobileNetV3.py如下所示 from torch import nn # ###### Mobilenetv3 class h_sigmoid(nn