ValueError: tf.function-decorated function tried to create variables on non-first call

问题:

tf 1x 的代码迁移到 tf 2x时出现错误:

ValueError: tf.function-decorated function tried to create variables on non-first call

解决方案:

在model.compiler()之后添加

model.run_eagerly = True

但是对tf 1x 定义的一些函数提示

WARNING:tensorflow:AutoGraph could not transform <function drop_connect at 0x0000024E55A0DA60> and will run it as-is.
Please report this to the TensorFlow team. When filing the bug, set the verbosity to 10 (on Linux, `export AUTOGRAPH_VERBOSITY=10`) and attach the full output.
Cause: module 'gast' has no attribute 'Index'
To silence this warning, decorate the function with @tf.autograph.experimental.do_not_convert

根据提示对函数,例如这里的drop_connect,前添加如下代码即可。

@tf.autograph.experimental.do_not_convert

 

  • 5
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值