discard distributed

CoinCheung · Jun 25, 2022 · 99951a1 · 99951a1
1 parent 628ebda
commit 99951a1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/lib/get_dataloader.py b/lib/get_dataloader.py
@@ -12,7 +12,7 @@
 
 
 
-def get_data_loader(cfg, mode='train', distributed=True):
+def get_data_loader(cfg, mode='train'):
     if mode == 'train':
         trans_func = T.TransformationTrain(cfg.scales, cfg.cropsize)
         batchsize = cfg.ims_per_gpu
@@ -28,7 +28,7 @@ def get_data_loader(cfg, mode='train', distributed=True):
 
     ds = eval(cfg.dataset)(cfg.im_root, annpath, trans_func=trans_func, mode=mode)
 
-    if distributed:
+    if dist.is_initialized():
         assert dist.is_available(), "dist should be initialzed"
         if mode == 'train':
             assert not cfg.max_iter is None

diff --git a/tools/train_amp.py b/tools/train_amp.py
@@ -121,10 +121,9 @@ def set_meters():
 
 def train():
     logger = logging.getLogger()
-    is_dist = dist.is_initialized()
 
     ## dataset
-    dl = get_data_loader(cfg, mode='train', distributed=is_dist)
+    dl = get_data_loader(cfg, mode='train')
 
     ## model
     net, criteria_pre, criteria_aux = set_model(dl.dataset.lb_ignore)