TensorRTOptimizer 警告

TensorRTOptimizer Warning

我每次优化张量流图时都会收到此警告:

TensorRTOptimizer is probably called on funcdef! This optimizer must *NOT* be called on function objects.

我只是想了解这个警告是什么意思,我该如何避免它以及它的影响。

我正在使用带有 cuda 10.0 和 tensorflow-gpu 1.15 的 tensorRT 5.1.5

谢谢!

忽略此警告是安全的。

警告信息在 TF 2.1 中有所改进,但它仍然存在。

简而言之,这是发生的事情:

我们将输入图命名为 tf_graph,然后在输入上调用 TF grappler。然后 grappler 对输入中可用的多个对象(包括 tf_graph)运行 TRT 优化传递。

TF-TRT 在内部检查正在使用的对象,只有在对象名称为 tf_graph 时才继续优化,否则打印警告消息和 returns.

复制 post: https://devtalk.nvidia.com/default/topic/1069865/tensorrt/tensorrtoptimizer-warning-tensorrtoptimizer-is-probably-called-on-funcdef-this-optimizer-must-not-be-called-on-function-objects-/