TensorRTOptimizer Warning

Question

Я получал это предупреждение каждый раз, когда оптимизировал график тензорного потока:

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

Я просто хочу понять, что означает это предупреждение, как я могу избежать его и каково его влияние.

Я использую tennorRT 5.1.5 с cuda 10.0 и tenorflow-gpu 1.15

Спасибо!

Answer 1

Можно игнорировать это предупреждение.

Предупреждение немного улучшено в TF 2.1, но оно все еще там.

Вот что происходит в двух словах:

Мы называем входной граф tf_graph и затем вызываем TF grappler на входе. Затем Grappler запускает этап оптимизации TRT для нескольких объектов, доступных во входных данных (включая tf_graph).

TF-TRT внутренне проверяет, какой объект используется, и продолжает оптимизацию, только если имя объекта равно tf_graph и в противном случае печатает предупреждающее сообщение и возвращает.

Dup 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-/