DeePTB
DeePTB copied to clipboard
Export the full trained PyTorch model to device-and-system-independent forms
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