美文网首页
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