Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to interface with python Torchrt models? #294

Open
karanamrahul opened this issue Jul 12, 2022 · 1 comment
Open

How to interface with python Torchrt models? #294

karanamrahul opened this issue Jul 12, 2022 · 1 comment

Comments

@karanamrahul
Copy link

No description provided.

@perseusdg
Copy link
Contributor

Unfortunately this repo doesn't support TorchRt models .You would need to manually export the weights from the pytorch model to layer wise weights and then write the model again using tkdnn layers,if your model has a cfg file like those found in yolo models and if it has only layers that are found in tkdnn , you can use the darknet parser to parse through the cfg file and return a model to you ,but you would still need to export the model weights to layer wise weights.You can have a look at the repos mentioned in exporting_weights.md file to get a general idea on how to export your model weigths to a tkdnn acceptable format and you can have a look at the example in the tests folder to get an idea on how to write the model using tkdnn layers or use the darknet parser.Hope this helps you out :)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants