Hi guys,
We at avyuct have found out how to use tensorRT model generated by TAO toolkit.
We were working on a classification problem using the resnet18 pretrained model and were running into issues when trying to deploy the tesnortRT model.
For all the folks who are stuck at the same phase, check out the attachments to see the configuration file which is used to genereate the tesortRT model and the python code to infer the tensorRT model.
tensorRT_model.py (3.0 KB)
classification_spec.cfg (1.2 KB)