PyNetsPresso
PyNetsPresso copied to clipboard
#245 Set download_model path extension to .pt
Description
Please include a summary in English, of the changes in this pull request. If it closes an issue, please mention it here.
Closes: #245
You should link at least one existing issue for PR. Before your create a PR, please check to see if there is an issue for this change.
PRs from forked repository not accepted.
Change(s)
- Set
download_modelpath extension to.pt - They are in
compress_model,recommendation_compressionandrecommendation_compressionmethods.
Sorry for reverting a commit.
I didn't recognized there is FRAMEWORK_EXTENSION_MAP.
I simply fixed FRAMEWORK_EXTENSION_MAP value, so that onnx file transformed to .pt