为了账号安全,请及时绑定邮箱和手机立即绑定

全局禁用 Tensorflow 断言?

全局禁用 Tensorflow 断言?

开心每一天1111 2021-10-19 09:46:36
Tensorflow 中是否有一个选项可以设置为忽略Assert我添加到我的所有操作Graph?我正在考虑的用例是使用 添加一堆断言tf.contrib.framework.with_shape,然后在我的配置文件中切换一个标志,config.fast以决定我是否应该在训练期间跳过这些断言。通常我会Asserts首先根据是否config.fast设置来选择是否生成,但是在使用时with_shape,将此选项链接到我的张量似乎更方便,而无需查看我的config.fast.
查看完整描述

1 回答

?
茅侃侃

TA贡献1842条经验 获得超21个赞

我最终使用了类似于@jdehesa 的评论建议的东西,但使用了分支工厂函数而不是tf.cond.


def make_maybe_with_shape(skip_asserts):

    if skip_asserts:  # identity

        return lambda expected_shape, tensor: tensor

    else:

        return lambda expected_shape, tensor: tf.contrib.framework.with_shape(

                expected_shape, tensor)


查看完整回答
反对 回复 2021-10-19
  • 1 回答
  • 0 关注
  • 169 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信