Skip to content
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

Merged
merged 9 commits into from
Jul 24, 2023

Conversation

eddogola
Copy link
Contributor

@eddogola eddogola commented Jul 6, 2023

Description

Update the mnist example to use the new compile_stage api

Checklist:

  • Has code been commented, particularly in hard-to-understand areas?

@eddogola eddogola changed the title initial setup mnist example update to new compile stage API Jul 6, 2023
@eddogola
Copy link
Contributor Author

eddogola commented Jul 6, 2023

@eddogola eddogola marked this pull request as draft July 6, 2023 18:56
@wz337 wz337 marked this pull request as ready for review July 7, 2023 00:34
@eddogola
Copy link
Contributor Author

eddogola commented Jul 7, 2023

@eddogola eddogola marked this pull request as draft July 10, 2023 18:37
@eddogola eddogola force-pushed the update-examples branch 2 times, most recently from ab5ba4f to 147c415 Compare July 10, 2023 18:54
@eddogola eddogola marked this pull request as ready for review July 20, 2023 23:01
@eddogola
Copy link
Contributor Author

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?

@wz337
Copy link
Contributor

wz337 commented Jul 20, 2023

Could you double check the files changed to make sure that the change is reflected correctly?

@wz337 wz337 marked this pull request as draft July 20, 2023 23:34
@eddogola eddogola marked this pull request as ready for review July 20, 2023 23:55
Copy link
Contributor

@wz337 wz337 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@eddogola eddogola merged commit 751b19d into pytorch:main Jul 24, 2023
@eddogola eddogola deleted the update-examples branch July 24, 2023 17:51
eddogola added a commit to eddogola/PiPPy that referenced this pull request Jul 25, 2023
## 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>
moonbucks pushed a commit to moonbucks/PiPPy that referenced this pull request Aug 1, 2023
## 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>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants