search

pytorch / ios-demo-app

PyTorch iOS examples
Other
599 stars 190 forks source link

"Failed to load model!" in project ObjectDetection #109

Open chelsea456 opened 1 year ago

chelsea456 commented 1 year ago

thank for you share project!! when i run the code , i have the prolem "Failed to load model!" although i have file yolov5s.torchscript.ptl. You can share me file yolov5s.torchscript.ptl because i think my file yolov5s.torchscript.ptl incorrect. tks sir !!!

jihyun-jeong-854 commented 1 year ago

Here is the link for the right one. https://pytorch-mobile-demo-apps.s3.us-east-2.amazonaws.com/yolov5s.torchscript.ptl

dmjtian commented 1 year ago

I have reproduced the demo, but I am experiencing this issue when using my own model.

jihyun-jeong-854 commented 1 year ago

We faced the same issue and we solved by using the exact same version of python, pytorch, cuda written at README when export torchscript file. Give it a try!!