Разве нельзя вычислить градиент из слоя 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?