How to input TorchScript model file (*.pt) and convert it to Torch MLIR directly (not python)
Hi, friends.
How to input TorchScript model file (*.pt) and convert it to Torch MLIR directly (not python) ?
I have implemented the above feature with python: import .pt file via Python’s torch.jit.load(model) and convert it to torch dialect MLIR via torch_mlir.compile().
Now I want to build a tool (or use torch-mlir-opt tool), how to input TorchScript model file (*.pt) and convert it to Torch MLIR directly?
thanks
AFAIK, there currently isn't a way to do this in torch-mlir.
AFAIK, there currently isn't a way to do this in torch-mlir.
Thanks
@imxxw Excuse me,I notice that you have implemented the above feature(input TorchScript model file (*.pt) and convert it to Torch MLIR directly) with python .Could I get your python script because i need to convert yoloV3.pt to MLIR.