PyNetsPresso icon indicating copy to clipboard operation
PyNetsPresso copied to clipboard

#245 Set download_model path extension to .pt

Open illian01 opened this issue 1 year ago • 1 comments

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 avatar May 03 '24 08:05 illian01

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

illian01 avatar May 03 '24 09:05 illian01