fix getenv

ultralytics · Jun 18, 2021 · 382ce4f · 382ce4f · glenn-jocher · Jun 18, 2021
1 parent fb342fc
commit 382ce4f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/train.py b/train.py
@@ -37,9 +37,9 @@
 from utils.wandb_logging.wandb_utils import WandbLogger, check_wandb_resume
 
 logger = logging.getLogger(__name__)
-LOCAL_RANK = int(getattr(os.environ, 'LOCAL_RANK', -1))  # https://pytorch.org/docs/stable/elastic/run.html
-RANK = int(getattr(os.environ, 'RANK', -1))
-WORLD_SIZE = int(getattr(os.environ, 'WORLD_SIZE', 1))
+LOCAL_RANK = int(os.getenv('LOCAL_RANK', -1))  # https://pytorch.org/docs/stable/elastic/run.html
+RANK = int(os.getenv('RANK', -1))
+WORLD_SIZE = int(os.getenv('WORLD_SIZE', 1))
 
 
 def train(hyp,  # path/to/hyp.yaml or hyp dictionary