Update configs.py—"LOCAL_RANK"

fairseq/fairseq/dataclass/configs.py

@@ -2,7 +2,7 @@
 #
 # This source code is licensed under the MIT license found in the
 # LICENSE file in the root directory of this source tree.
-
+import os
 import sys
 from dataclasses import _MISSING_TYPE, dataclass, field
 from typing import Any, List, Optional
@@ -271,12 +271,18 @@ class DistributedTrainingConfig(FairseqDataclass):
         },
     )
     device_id: int = field(
-        default=0,
+        default=int(os.environ["LOCAL_RANK"]),
         metadata={
             "help": "which GPU to use (usually configured automatically)",
             "argparse_alias": "--local_rank",
         },
     )
+    local_rank: int = field(
+        default=int(os.environ["LOCAL_RANK"]),
+        metadata={
+            "help": "which GPU to use (usually configured automatically)",
+        },
+    )
     distributed_no_spawn: bool = field(
         default=False,
         metadata={