machinelearning icon indicating copy to clipboard operation
machinelearning copied to clipboard

Loading a LSTM Model Created in Torchsharp in ML.Net

Open aforoughi1 opened this issue 1 year ago • 0 comments

I can create/train/evaluate/save/load a multilayer LSTM model using Torchsharp but cannot run in ML.NET.

The main requirement is to export/import a Torchsharp model to ONNX and run it in ML.NET. However, Torchsharp doesn't implement torch.onnx.export method. I'm stuck at a point and had to convert everything to pytorch just to export to ONNX.

I also tried a custom pipeline and trial runner to train and evaluate the model but I'm struggling with how these models can be saved/Loaded in ML.NET.

aforoughi1 avatar May 29 '24 10:05 aforoughi1