Skip to content

Add example program of MNIST using pytorch DDP #43

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

Merged
merged 20 commits into from
Aug 12, 2024

Conversation

KWang1998
Copy link
Collaborator

@KWang1998 KWang1998 commented Aug 10, 2024

MNIST Pytorch program is obtained from Pytorch/MNIST
A code patch mnist.patch must be applied first.

Command make check automatically downloads, applies the patch,
and run the training with small settings, i.e.

  • train batch size = 4,
  • test batch size = 2,
  • number of epochs = 3,
  • number of training samples = 60, and
  • number of testing samples = 12

@wkliao wkliao merged commit 995a2b1 into Parallel-NetCDF:main Aug 12, 2024
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants