Tensorflow如何对VGG19做L2正则化

2017-10-26  本文已影响79人  小王子威威

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

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

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

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

Tensorflow中具体实施如下:

在后面主函数里:

上一篇 下一篇

猜你喜欢

热点阅读