search

chufengt / ALM-pedestrian-attribute

Code for the paper "Improving Pedestrian Attribute Recognition With Weakly-Supervised Multi-Scale Attribute-Specific Localization", ICCV 2019, http://arxiv.org/abs/1910.04562.
Apache License 2.0
188 stars 66 forks source link

TensorRT model conversion #34

Closed abhigoku10 closed 3 years ago

abhigoku10 commented 3 years ago

@chufengt can we convert the current pretrained or custom model into tensor rt for faster inference ?

chufengt commented 3 years ago

Sorry, I have no experience with Tensor RT. You can have a try.

A7med01 commented 3 years ago

@abhigoku10 did you successfully convert the model to tensor rt??

abhigoku10 commented 3 years ago

@chufengt nope i did not go to that phase of conversion , do let me knw if your able to convert

abhigoku10 commented 3 years ago

@chufengt @A7med01 i am getting this below error when converting to onnx "RuntimeError: Exporting the operator affine_grid_generator to ONNX opset version 11 is not supported. Please feel free to request support or submit a pull request on PyTorch GitHub"

abhigoku10 commented 3 years ago

solved in #44