美文网首页
Tensorflow如何对VGG19做L2正则化

Tensorflow如何对VGG19做L2正则化

作者: 小王子威威 | 来源:发表于2017-10-26 17:18 被阅读79次

在Tensorflow下用VGG19 pre-train的model跑一个人脸表情库,做一个二分类。

出现loss除了迭代的第一个值,其余输出均是0.693174

百度解决方法,发现需要在全连接层的权重加权重衰减(l2正则化)

在把权重衰减loss和交叉熵loss添加到总loss里。

Tensorflow中具体实施如下:

在后面主函数里:

相关文章

网友评论

      本文标题:Tensorflow如何对VGG19做L2正则化

      本文链接:https://www.haomeiwen.com/subject/dtbkpxtx.html