#245 Set download_model path extension to .pt

Nota-NetsPresso / PyNetsPresso

The official NetsPresso Python package.

https://nota-netspresso.github.io/PyNetsPresso/

Apache License 2.0

40 stars 1 forks source link

#245 Set download_model path extension to .pt #247

Closed illian01 closed 3 months ago

illian01 commented 5 months ago

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_model path extension to .pt
They are in compress_model, recommendation_compression and recommendation_compression methods.

illian01 commented 5 months ago

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