zoukankan      html  css  js  c++  java
  • tensorflow中添加L2正则化损失(Tensorflow v1.x 版本)

    方法有几种,总结一下方便后面使用。

    1. tensorflow自动维护一个tf.GraphKeys.WEIGHTS集合,手动在集合里面添加(tf.add_to_collection())想要进行正则化惩罚的变量。

    然后创建 regularizer = tf.contrib.layers.l2_regularizer(REGULARIZATION_RATE),

    再应用函数 regularization_loss = tf.contrib.layers.apply_regularization(regularizer=regurializer) 即可得到对集合tf.GraphKeys.WEIGHTS内的变量的正则化项。


    2. 先创建 regularizer =  tf.contrib.layers.l2_regularizer(REGULARIZATION_RATE),

    创建变量时指定regularizer,如 w1 = tf.get_variable('w1', [100, 100], regularizer=regularizer),tensorflow会将变量加入集合 tf.GraphKeys.REGULARIZATOIN_LOSSES,

    然后设置正则化系数 REGULARIZATION_RATE,通过获取上述的集合即可得到正则化损失

    regularization_loss = REGULARIZATION * sum(tf.get_collection(tf.GraphKeys.REGULARIZATION_LOSSES))。

  • 相关阅读:
    node异步转同步(循环)
    三级省市区PCASClass.js插件
    微信公众号基础总结(待更新)
    ES6详解
    webpack配置
    高性能 CSS3 动画
    github上传口令
    纯css3 实现3D轮播图
    优美的js代码,拿去玩~
    关于列举属性用点还是用【】
  • 原文地址:https://www.cnblogs.com/banluxinshou/p/9799969.html
Copyright © 2011-2022 走看看