GitHubEmploy/pytorch_8.py

Created December 30, 2021 21:19

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/GitHubEmploy/3991a9b8269bcf16275eaf5d76457a6e.js"></script>
Save GitHubEmploy/3991a9b8269bcf16275eaf5d76457a6e to your computer and use it in GitHub Desktop.

Download ZIP

Raw

pytorch_8.py

torch.save(model.state_dict(), 'model.ckpt')

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment