全局禁用 Tensorflow 断言?

Tensorflow 中是否有一个选项可以设置为忽略Assert我添加到我的所有操作Graph

我正在考虑的用例是使用 添加一堆断言tf.contrib.framework.with_shape,然后在我的配置文件中切换一个标志,config.fast以决定我是否应该在训练期间跳过这些断言。

通常我会Asserts首先根据是否config.fast设置来选择是否生成,但是在使用时with_shape,将此选项链接到我的张量似乎更方便,而无需查看我的config.fast.


开心每一天1111
浏览 170回答 1
1回答

茅侃侃

我最终使用了类似于@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)
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python