一种模型部署方法和装置
    6.
    发明公开

    公开(公告)号:CN116776936A

    公开(公告)日:2023-09-19

    申请号:CN202310712796.2

    申请日:2023-06-15

    摘要: 本发明公开了一种模型部署方法和装置,所述方法包括:获取预先训练完成的视觉卷积模型,设置对视觉卷积模型进行剪枝操作的精度目标和搜索参数,搜索参数至少包括初始剪枝比例和搜索区间长度;分别搜索注意力机制模块的第一目标剪枝比例和前馈神经模块的第二目标剪枝比例;根据第一目标剪枝比例和第二目标剪枝比例优化视觉卷积模型,得到目标视觉卷积模型;将目标视觉卷积模型部署在变电站的边缘计算设备上。这样,在对视觉卷积模型进行剪枝后,使得模型大小处于最小的状态,从而对运行环境的要求降低。将视觉卷积模型部署在边缘计算设备上,实现了监控数据的实时接收,从而可以做到实时的检测和警示,提高了电力系统的安全性。