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

Minimal harness for running CIFAR10 experiments #9

Open
wants to merge 3 commits into
base: main
Choose a base branch
from

Conversation

nelaturuharsha
Copy link

Hello! Just in case it might be useful for others who'd like to test the optimizer out as well on the CIFAR10 dataset -- just wanted to add my training harness.

Should be easy to swap out out with any other model, or something like CIFAR100 as well.

Its more or less in the same spirit as the MNIST example, and should be easily runnable :)

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Meta Open Source bot. label Apr 6, 2024
@adefazio
Copy link
Contributor

adefazio commented Apr 7, 2024

Thanks! I will look over this Monday.

@adefazio
Copy link
Contributor

adefazio commented Apr 9, 2024

Thanks for the contribution. We are going to decided on the exact ResNet-18 variant and default training parameters over the next few days.

@nelaturuharsha
Copy link
Author

Sounds good :) Thanks for looking into it -- happy to modify if required!

@tfaod
Copy link
Contributor

tfaod commented Apr 17, 2024

@nelaturuharsha I'm currently working on this, and will have this out soon

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Meta Open Source bot.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants