DeePTB icon indicating copy to clipboard operation
DeePTB copied to clipboard

Export the full trained PyTorch model to device-and-system-independent forms

Open AlexanderZ11234 opened this issue 2 months ago • 0 comments

Background

By "device-and-system-independent forms", I mean a full model (not just its trained parameters) stored in a format that can be imported and used by programming languages other than Python. The format can, but is not limited to, e.g., *.onnx.

Describe the solution you'd like

The present stored model is a *.pth file, which I am not sure is the full model or just the trained parameters. It would be highly appreciated if in just one line of command the *.pth file is transformed into a *.onnx file.

Additional Context

No response

AlexanderZ11234 avatar Apr 26 '24 07:04 AlexanderZ11234