I am referring below link for converting .pt to .ptl(PyTorch lite), Please check the code and please let me know if I am doing anything wrong - (Prototype) Introduce lite interpreter workflow in Android and iOS — PyTorch Tutorials 1.9.0+cu102 documentation
here is my code -
import torch
model = torch.hub.load(‘ultralytics/yolov5’, ‘custom’,path=‘/content/drive/MyDrive/yolov5/runs/train/exp51/weights/last.pt’)
model.eval()
scripted_module = torch.jit.script(model)
scripted_module.save(“/content/drive/MyDrive/yolov5/runs/train/exp51/weights/last.pt”)
scripted_module._save_for_lite_interpreter(“/content/drive/MyDrive/yolov5/runs/train/exp51/weights/last.ptl”)
Error screenshot -
Is there any other way to convert the .pt file into .ptl?