pytorch/ios-demo-app

let module = TorchModule(fileAtPath: filePath) failed

QinZiwen opened this issue · 4 comments

QinZiwen commented 
pip install torchvision
python trace_model.py
pod install

work well, but when I select an iOS simulator and launch it, "let module = TorchModule(fileAtPath: filePath)" failed to run. I can be sure that filePath is right.

thanks

xta0 commented

@jeffxtang could you help take a look?

jeffxtang commented

@QinZiwen what's the error message you got?

QinZiwen commented

@jeffxtang @xta0 thanks
It was found that CUDA was set when the model was exported. After deletion, everything was normal

  • 👍2
CarlosModinez commented

@QinZiwen could you explain what you did? I'm struggling with the same problem.
Thanks