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

转换自己的torch模型报错:RuntimeError: PytorchStreamReader failed locating file constants.pkl: file not found #178

Open
dzy1128 opened this issue Jul 25, 2024 · 1 comment

Comments

@dzy1128
Copy link

dzy1128 commented Jul 25, 2024

转换代码为:model_transform --model_name yolov8_pt --model_def ../best.pt --input_shapes [[1,3,640,640]] --mean 0.0,0.0,0.0 --scale 0.0039216,0.0039216,0.0039216 --keep_aspect_ratio --pixel_format rgb --test_input ../traffic-images/00482.jpg --test_result yolov8_pt_top_outputs.npz --mlir yolov8_pt.mlir
6b8325bb6df02adb39a3475e173b3d1f

@lordrebel
Copy link

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants