一种VGG网络的训练方法及装置与流程

文档序号:12964267阅读:来源:国知局

技术特征:

技术总结
本发明公开了一种VGG网络的训练方法及装置,其中该方法包括:生成包含有n个全连接层的VGG网络模型;对所述VGG网络模型进行训练,并在每步训练后将每个所述全连接层的权重矩阵中最接近于零的元素置零,直至每个所述全连接层中为零的元素在该全连接层的全部元素中所占比例达到预设值为止。本申请公开的上述技术方案中,实现了对全连接层参数的逐步精简,每次精简只丢失表现最弱的那个权重元素,从而不仅能够有效的防止重要权重的丢失,有效防止精度下降,还可以使得VGG网络模型更加精简,使得训练得到的VGG网络模型所占内存较少,进而能够移植到移动设备中。

技术研发人员:蔡述庭;梁天智;陈平;李卫军;刘坤
受保护的技术使用者:广东工业大学
技术研发日:2017.07.26
技术公布日:2017.11.21
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1