- 博客(2)
- 收藏
- 关注
RSS订阅原创 tf.constrib.slim计算损失函数
Tensorflow的模型训练需要模型，损失函数，梯度计算，以及根据loss的梯度迭代更新参数,使用tf.constrib.slim很方便。（1）losses使用现有的loss：loss = slim.losses.softmax_cross_entropy(predictions, labels) 对于多任务学习的loss，可以使用： # Define the loss functions an...
2018-03-29 10:34:26
965
原创 tensorflow L2正则化
L2正则化在机器学习和深度学习非常常用，在TensorFlow中使用L2正则化非常方便，仅需将下面的运算结果加到损失函数后面即可reg = tf.contrib.layers.apply_regularization(tf.contrib.layers.l2_regularizer(1e-4), tf.trainable_variables())...
2018-03-17 18:19:05
4011
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人