deepmodeling / deepmodeling/DeePTB

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

Open
#143 0 comments 0 reactions 0 assignees View on GitHub
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

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.