bug for distributed training (open-mmlab#1985)

fix a bug for distributed training in windows platform
ioir123ju · Mar 30, 2020 · 101a76d · 101a76d
1 parent 1384463
commit 101a76d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmdet/apis/train.py b/mmdet/apis/train.py
@@ -73,7 +73,7 @@ def parse_losses(losses):
     log_vars['loss'] = loss
     for loss_name, loss_value in log_vars.items():
         # reduce loss when distributed training
-        if dist.is_initialized():
+        if dist.is_available() and dist.is_initialized():
             loss_value = loss_value.data.clone()
             dist.all_reduce(loss_value.div_(dist.get_world_size()))
         log_vars[loss_name] = loss_value.item()