APP下载
[00065483]一种VGG网络的训练方法及装置
联系人： 厦门立德软件公司
所在地：福建 厦门市
- 服务承诺
- 产权明晰
-
资料保密
对所交付的所有资料进行保密
- 如实描述
技术详细介绍
本发明公开了一种VGG网络的训练方法及装置，其中该方法包括生成包含有n个全连接层的VGG网络模型；对所述VGG网络模型进行训练，并在每步训练后将每个所述全连接层的权重矩阵中最接近于零的元素置零，直至每个所述全连接层中为零的元素在该全连接层的全部元素中所占比例达到预设值为止。本申请公开的上述技术方案中，实现了对全连接层参数的逐步精简，每次精简只丢失表现最弱的那个权重元素，从而不仅能够有效的防止重要权重的丢失，有效防止精度下降，还可以使得VGG网络模型更加精简，使得训练得到的VGG网络模型所占内存较少，进而能够移植到移动设备中。
