correction for get rank id. (k2-fsa#507)

* Fix no attribute 'data' error. * minor fixes * correction for get rank id.
ZuoyunZheng · Jul 29, 2022 · 34b4356 · 34b4356
1 parent ec69967
commit 34b4356
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/icefall/dist.py b/icefall/dist.py
@@ -57,9 +57,9 @@ def get_rank():
     if "RANK" in os.environ:
         return int(os.environ["RANK"])
     elif dist.is_available() and dist.is_initialized():
-        return dist.rank()
+        return dist.get_rank()
     else:
-        return 1
+        return 0
 
 
 def get_local_rank():