deepmodeling / deepmodeling/DeePTB
Export the full trained PyTorch model to device-and-system-independent forms
Open
- Dominant language
- Python
- Stars
- 122
- Forks
- 36
- PR merge metrics
- No merged PRs in 30d
Description
### 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_
Contributor guide
Assessment
This issue has not been assessed yet.