We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e032ebb commit 4b984a7Copy full SHA for 4b984a7
mmdet/apis/train.py
@@ -73,7 +73,7 @@ def parse_losses(losses):
73
log_vars['loss'] = loss
74
for loss_name, loss_value in log_vars.items():
75
# reduce loss when distributed training
76
- if dist.is_initialized():
+ if dist.is_available() and dist.is_initialized():
77
loss_value = loss_value.data.clone()
78
dist.all_reduce(loss_value.div_(dist.get_world_size()))
79
log_vars[loss_name] = loss_value.item()
0 commit comments