-
Notifications
You must be signed in to change notification settings - Fork 3.4k
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
[WIP] ref: part 6 of #3733 #3783
Conversation
if is_master: | ||
available_gpus = os.environ['CUDA_VISIBLE_DEVICES'].split(',') | ||
gpu_idx = int(available_gpus[self.trainer.local_rank]) | ||
gpu_idx = int(os.environ.get('PL_DDP_PID', process_idx)) |
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.
is_master
is never used here. it seems like ddp_train doesn't need the self.mode == 'ddp' check anymore
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.
Good catch!
might be a weird merge artifact... let me check.
if 'WORLD_SIZE' in os.environ: | ||
del os.environ['WORLD_SIZE'] |
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.
n00b question: should this happen in a teardown step?
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.
probably haha.
i'm not sure what will happen if we move it though. I want to lock down the tests then we can see where it goes
Hello @williamFalcon! Thanks for updating this PR. There are currently no PEP 8 issues detected in this Pull Request. Cheers! 🍻 Comment last updated at 2020-10-02 11:15:22 UTC |
Codecov Report
@@ Coverage Diff @@
## master #3783 +/- ##
========================================
- Coverage 85% 43% -42%
========================================
Files 110 110
Lines 8537 8582 +45
========================================
- Hits 7293 3708 -3585
- Misses 1244 4874 +3630 |
ref: part 6 of #3733