Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- torch.save(model.state_dict(), 'model.ckpt')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement