Можно ли получить градиент из слоя TRTEngineOp? - PullRequest
0 голосов
/ 17 июня 2020

Разве нельзя вычислить градиент из слоя TRTEngineOp, который преобразован с помощью tensorflow-tensorRT (tf-trt)?

Я хочу посмотреть, есть ли разница в визуализации GradCAM между вес исходной предварительно обученной модели и преобразованная модель, которая содержит слой TRTEngineOp с использованием tensorRT.

Для визуализации GradCAM преобразованный TRTEngineOp необходим для слоев, даже для слоя TRTEngineOp. Но когда я использую tf.gradient, возникает следующая ошибка LookupError.

LookupError: No gradient defined for operation'import/efficientnetb0/TRTEngineOp_0' (op type: TRTEngineOp)

Невозможно ли вычисление градиента в слое, преобразованном в TRTEngineOp с помощью TensorRT?

...