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)
添加回答
举报
0/150
提交
取消