-
Notifications
You must be signed in to change notification settings - Fork 88
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
mnist example update to new compile stage API #831
Conversation
See error here: https://www.internalfb.com/phabricator/paste/view/P782391812 |
See new error here: https://www.internalfb.com/phabricator/paste/view/P784276221 |
ab5ba4f
to
147c415
Compare
I'm currently saving the ckpts for each epoch in its own directory s.t. if were saving for every 500 epochs, there is a dir under "checkpoints"(or whatever other string that may be set as the checkpoint_dir) called 500, another 1000, and so on. What do you think of this design choice? |
Could you double check the files changed to make sure that the change is reflected correctly? |
a6a2b24
to
0115743
Compare
with compile_stage one
0115743
to
a6ddbe6
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM!
## Description Update the mnist example to use the new compile_stage api ## Checklist: - [x] Has code been commented, particularly in hard-to-understand areas? --------- Co-authored-by: Eddy <ogolaedd@meta.com> Co-authored-by: Eddy Ogola Onyango <ogolaedd@fb.com>
## Description Update the mnist example to use the new compile_stage api ## Checklist: - [x] Has code been commented, particularly in hard-to-understand areas? --------- Co-authored-by: Eddy <ogolaedd@meta.com> Co-authored-by: Eddy Ogola Onyango <ogolaedd@fb.com>
Description
Update the mnist example to use the new compile_stage api
Checklist: