Skip to content

Commit

Permalink
test (#2341)
Browse files Browse the repository at this point in the history
* fixes rank zero issue
  • Loading branch information
williamFalcon committed Jun 24, 2020
1 parent a915280 commit c09b2ff
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions pytorch_lightning/trainer/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -327,8 +327,6 @@ def __init__(
# this way we only show it on rank 0
if 'LOCAL_RANK' in os.environ:
rank_zero_only.rank = os.environ['LOCAL_RANK']
if 'SLURM_LOCALID' in os.environ:
rank_zero_only.rank = os.environ['SLURM_LOCALID']

# Init callbacks
self.prepare_data_per_node = prepare_data_per_node
Expand Down

0 comments on commit c09b2ff

Please sign in to comment.