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

[Update] Print training log in rank0 #1296

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

shijungg
Copy link

Now the training log is printed in rank_last, it's not friendly to users if they change the amount of GPU node.

For example, a multi-node training job has master-0, worker-0, worker-1...
If we use 32 node(8 gpus each), we need to find the training log in worker-30.
Next time we use 64 node(8 gpus each), we need to find the training log in worker-62.

So print training log in rank0 is more friendly, we can just find the training log in master-0, no mater how many nodes we use for training.

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.

1 participant