tensorflow旧版本0.11的初始化问题 zeros_initializer() got multiple values for argument……

立个flag,找个工具把代码升级成1.0之后的版本,尽快。

使用tf.get_variable时,initializer的问题

使用zeros_initializerones_initializer时,不要加小括号,那是新版本的用法

根据网上的答案,将zeros_initialier换成constant_initialier(0.0)依然报错

去掉括号之后依然有问题,但多试了几遍就不报错了…… 不解

我不知道为什么,多试了几遍就好了……十分不解

所以还是,换新版本。旧版本好多问题查不到。

参考https://github.com/ry/tensorflow-resnet/blob/master/resnet.py,实现微调resnet50时的bn层时,遇到的问题

略………………
moving_mean = _get_variable('moving_mean',
                                params_shape,
                                initializer=tf.zeros_initializer,
                                trainable=False)
moving_variance = _get_variable('moving_variance',
                                    params_shape,
                                    initializer=tf.ones_initializer,
                                    trainable=False)

_get_variable调用了tf.get_variable

def _get_variable(name,
                  shape,
                  initializer,
                  weight_decay=0.0,
                  dtype='float',
                  trainable=True):
    "A little wrapper around tf.get_variable to do weight decay and add to"
    "resnet collection"
    if weight_decay > 0:
        regularizer = tf.contrib.layers.l2_regularizer(weight_decay)
    else:
        regularizer = None
    collections = [tf.GraphKeys.VARIABLES, RESNET_VARIABLES]
    return tf.get_variable(name,
                           shape=shape,
                           initializer=initializer,
                           dtype=dtype,
                           regularizer=regularizer,
                           collections=collections,
                           trainable=trainable)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值