github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-02 17:46:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Gao, Ruiyuan
2024-07-24 10:30:40 +08:00
committed by GitHub
parent a521ffc9f8
commit 5fb958cc83
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/cluster/dist_coordinator.py
View File

@@ -44,7 +44,7 @@ class DistCoordinator(metaclass=SingletonMeta):
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: