I have to train model with ‘x’ outputs . Here each output use loss ‘categorical_cross_entropy’ and activation ‘softmax’. Is this possible by using TLT? How to convert the multi output model into TensorRT?

Please note that currently TLT classification network does not support multioutput training. But it is on the roadmap.

Train and convert Multioutput model into TensorRT by using TLT 2.0

Accelerated Computing Intelligent Video Analytics TAO Toolkit

samjith888 August 20, 2020, 7:42am 7

But u have mentioned in the following thread that , The Tensorflow multi output model to UFF convertion is possible. Hence the UFF to TensorRT is also possible right ?