当前位置: 首页 > news >正文

卷积通用模型的剪枝、蒸馏---剪枝篇(此处以deeplabv3+为例,可根据模型自行定制剪枝层)

之后的两篇文章是对前段时间工作的一个总结。
一、环境配置
1.1、文章以b导的代码为模板,环境配置比较简单(第二篇蒸馏篇结束后会放置剪枝蒸馏配置好的百度网盘链接),其他算法自行配置,在剪枝之前,需要保证算法能够在本地跑通。
B导链接:
https://github.com/bubbliiiing/deeplabv3-plus-pytorch
1.2、安装torch_pruning
在利用剪枝功能前,需要安装一下剪枝的库。需要安装0.2.7版本,剪枝时的一些log信息会自动保存在logs文件夹下。

pip install torch_pruning==0.2.7

在这里插入图片描述
二、剪枝过程
2.1、读取模型结构
此处分两种情况,第一种为torch.save(model,…)保存的权重,第二种为torch.save(model.state_dict(),…)保存的权重,如下适用于第二种情况下,第一种情况本身包含模型结构信息。

def 
http://www.lryc.cn/news/335846.html

相关文章:

  • 使用Ollama在本地运行AI大模型gemma
  • 【IC前端虚拟项目】时序面积优化与综合代码出版本交付
  • windows版本-idea中下载的java版本在哪
  • 设计模式:创建者模式
  • 【linux】基础IO(四)
  • 集合框架(数组,Arrays.sort,list,map,set,stack,queue)蓝桥杯习题
  • 【C++基础】运算符和流程控制语句
  • AOF文件重写
  • 第四次面试总结 — 嘉和智能 - 全栈开发
  • tx-lcn使用
  • oracle恢复异常处理
  • 谈谈什么是 Redis
  • 备战蓝桥杯(java)(日益更新)
  • 06 Php学习:字符串
  • python画图Matplotlib和Seaborn
  • 一体式I/O模块与RS485串口联动,实现工业网络无缝对接
  • 如何在Uniapp真机中使用H.265无插件流媒体播放器EasyPlayer
  • 【MySQL数据库 | 第二十五篇】深入探讨MVCC底层原理
  • 【LAMMPS学习】八、基础知识(1.6) LAMMPS 与其他代码耦合
  • ArrayList中多线程的不安全问题
  • ALTER TABLE 之 定点变更(in-place alter)
  • 【RAG实践】Rerank,让大模型 RAG 更近一步
  • 私有化客服系统:在线客服搭建与部署的创新之路
  • VM-UNet: Vision Mamba UNet for Medical Image Segmentation
  • 面向对象编程:在Python中的面向对象编程奥秘
  • 考研数学|零基础100分保底复习方案+资料分享
  • 【MATLAB源码-第29期】基于matlab的MIMO,MISO,SIMO,SISO瑞利rayleigh信道容量对比。
  • JRT高效率开发
  • Spring Boot 切面的一种的测试方法,java中级开发面试
  • 嵌入式自学路线-高薪路线(持续更新,欢迎关注)