logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

模型剪枝-Network Slimming

模型剪枝是一种优化神经网络的技术,通过移除不重要的权重或神经元来降低模型的复杂度,同时尽量保持其性能。本文基于经典论文《Learning Efficient Convolutional Networks through Network Slimming》实现模型剪枝,主要步骤包括训练模型、剪枝和再次训练。剪枝过程中,BatchNorm(BN)和L1正则化是关键。BN通过标准化特征图,缓解内部协变量

#剪枝#算法#机器学习
MMPretrain的使用以及训练自己的数据集

MMPretrain是一个基于PyTorch的深度学习框架,专注于图像分类任务。用户可以通过GitHub下载其压缩包,并使用pip命令安装相关依赖。在配置文件中,用户可以选择并修改预训练模型(如resnet18_8xb32_in1k.py)以适应特定任务,例如将1000类分类改为102类花朵分类。通过自定义数据集加载代码(如myimagenet.py),用户可以加载自己的数据集并进行训练。MMPr

文章图片
#python
MMPretrain的使用以及训练自己的数据集

MMPretrain是一个基于PyTorch的深度学习框架,专注于图像分类任务。用户可以通过GitHub下载其压缩包,并使用pip命令安装相关依赖。在配置文件中,用户可以选择并修改预训练模型(如resnet18_8xb32_in1k.py)以适应特定任务,例如将1000类分类改为102类花朵分类。通过自定义数据集加载代码(如myimagenet.py),用户可以加载自己的数据集并进行训练。MMPr

文章图片
#python
到底了