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

[PaddleV3] 修复 pytorchaten_linear 的算子映射并修复相关模型 #1079

Merged
merged 5 commits into from
Nov 1, 2024

Conversation

megemini
Copy link
Contributor

Create A Good Pull Request

修复 pytorchaten_linear 的算子映射并修复相关模型 ~

aten_linear 中的 bias 会生成为 None ,导致模型运行失败 ~

修改为:根据 weightdtypeshape(1,) 生成一个值为 0bias Tensor ~

涉及模型:

  • test_benchmark/PyTorch/MiniFasNet
  • test_benchmark/PyTorch/SwinTransformer

@PaddlePaddle PaddlePaddle locked and limited conversation to collaborators Oct 23, 2024
@PaddlePaddle PaddlePaddle unlocked this conversation Oct 23, 2024
@luotao1 luotao1 added the contributor External developers label Oct 23, 2024
@megemini
Copy link
Contributor Author

Update 20241031

  • 更新 black.list ,CI 通过测试 ~
  • SwinTransformer torch 加载模型时 map_location 到 CPU,还是之前环境 GPU 的问题 ... ...

@luotao1

@luotao1 luotao1 merged commit cfa1300 into PaddlePaddle:develop Nov 1, 2024
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
contributor External developers
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants