【发布时间】:2018-06-20 16:32:31
【问题描述】:
我正在使用tf.losses.sparse_softmax_cross_entropy 作为最后一层的分类函数来训练我的网络,一切正常。
我现在只是在我的权重上添加了一个 L2 正则化,我的损失不再得到优化。会发生什么?
reg = tf.nn.l2_loss(w1) + tf.nn.l2_loss(w2)
loss = tf.reduce_mean(tf.losses.sparse_softmax_cross_entropy(y, logits)) + reg*beta
train_step = tf.train.GradientDescentOptimizer(learning_rate).minimize(loss)
【问题讨论】:
-
你试过缩小你的 l2 损失吗?
标签: python tensorflow neural-network loss-function