Upload folder using huggingface_hub

logger.py

@@ -13,8 +13,7 @@ from collections import defaultdict, deque
 import torch
 import torch.distributed as dist
 
-import dist_utils
-
+from .dist_utils import is_dist_avail_and_initialized,is_main_process
 
 class SmoothedValue(object):
     """Track a series of values and provide access to smoothed values over a
@@ -38,7 +37,7 @@ class SmoothedValue(object):
         """
         Warning: does not synchronize the deque!
         """
-        if not dist_utils.is_dist_avail_and_initialized():
+        if not is_dist_avail_and_initialized():
             return
         t = torch.tensor([self.count, self.total], dtype=torch.float64, device="cuda")
         dist.barrier()
@@ -189,7 +188,7 @@ class AttrDict(dict):
 
 def setup_logger():
     logging.basicConfig(
-        level=logging.INFO if dist_utils.is_main_process() else logging.WARN,
+        level=logging.INFO if is_main_process() else logging.WARN,
         format="%(asctime)s [%(levelname)s] %(message)s",
         handlers=[logging.StreamHandler()],
     )