-
Notifications
You must be signed in to change notification settings - Fork 9.7k
Add distributed usage to ImageNet example's README #1001
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
base: main
Are you sure you want to change the base?
Conversation
✅ Deploy Preview for pytorch-examples-preview ready!
To edit notification comments on pull requests, go to your Netlify site settings. |
python main.py ... --world-size 2 --rank 0 --gpu 0 [imagenet-folder with train and val folders] & | ||
python main.py ... --world-size 2 --rank 1 --gpu 1 [imagenet-folder with train and val folders] & |
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.
@Quentin-Anthony noob question: why do you need the trailing "&"?
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.
@hudeven -- It launches the line in the background so that bash can run the next command immediately
I think the print for |
I'm confused. This PR is for updating the README. What's being printed? |
Following the discussions in #995 and #862, This PR adds a small usage example into the ImageNet README.
One note, however, is that specifying
--gpu
will trigger the print in https://github.com/pytorch/examples/blob/main/imagenet/main.py#L95-L97, which is confusing to users. Should we either remove or clarify this print?