[FIX BUG] convert env param to int in (#5934)

hpcaitech · Jul 24, 2024 · 5fb958c · 5fb958c
1 parent a521ffc
commit 5fb958c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/colossalai/cluster/dist_coordinator.py b/colossalai/cluster/dist_coordinator.py
@@ -44,7 +44,7 @@ def __init__(self):
         self._rank = dist.get_rank()
         self._world_size = dist.get_world_size()
         # this is often passed by launchers such as torchrun
-        self._local_rank = os.environ.get("LOCAL_RANK", -1)
+        self._local_rank = int(os.environ.get("LOCAL_RANK", -1))
 
     @property
     def rank(self) -> int: