distributed_tutorial icon indicating copy to clipboard operation
distributed_tutorial copied to clipboard

Published 20 hours ago • verified publisher icon yangkky

save or load checkpoint

Open rancheng opened this issue 4 years ago • 0 comments

https://github.com/yangkky/distributed_tutorial/blob/24467967c1c719110c33fccca69353ad8e5ae2e4/src/mnist-mixed.py#L108-L114

could you add the save model line to the example to be more complete. Thanks!

torch.save(model.state_dict(), CHECKPOINT_PATH)

rancheng avatar Nov 10 '20 04:11 rancheng