转换规则 torch.hub.load_state_dict_from_url #226
Open
+87
−1
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
Thanks for your contribution! |
|
torch.utils.model_zoo.load_url _test_alias_case_1()在ci上的pytorch版本已去掉，运行ci会报错，因此单测加_跳过，提供本地运行截图 |
在支持转换的参数中，是一致的 |
zhwesky2010
reviewed
Aug 18, 2023
| "unsupport_args": [ | ||
| "model_dir", | ||
| "map_location", | ||
| "progress", |
There was a problem hiding this comment.
这些不重要的参数可以直接删除，避免影响转换效果，一般一些只关乎性能的、调试的、输入校验的，对计算结果都没有影响，可以采取这种描述方式，代码里写法是对应一个空字符串，文档记得同步更新
这里的progress、check_hash应该不关键
| pytorch_code = textwrap.dedent( | ||
| """ | ||
| import torch | ||
| state_dict = torch.hub.load_state_dict_from_url('https://s3.amazonaws.com/pytorch/models/resnet18-5c106cde.pth') |
There was a problem hiding this comment.
看了下这两个API的返回结果不同，torch返回的是一个Layer的state_dict结果，paddle返回的是一个路径，所以是不是paddle的还需要包装下，比如加一个load之类的
There was a problem hiding this comment.
加load之后还是不能实现直接转换，原因是：torch 的.pth权重文件，paddle直接load .pth可能会报错
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR Docs
PaddlePaddle/docs#6090
PR APIs