Hello. I am trying to convert a trained model using the TAO toolkit. My question is when I export the model (whether its fp32 or int8) the output layer before changes from ‘softmax_1’ to ‘argmax_1’ and the output class also changes from (None, 544, 960, 2) to (None, 544, 960, 1) I need to know what…

That’s expected behavior. In the training config, the activation will be used on the last layer. The supported activation is sigmoid or softmax. If use softmax, the output layer would be replaced with argmax by TAO for optimization. Hence, you need to output-blob-names=argmax_1 in deepstream. Mor…

TAO-Export changes output layers

Accelerated Computing Intelligent Video Analytics TAO Toolkit

lucasp July 25, 2022, 4:05pm 14

Link to new topic for anyone who wants to follow the topic:

Cheers