Skip to content

Commit e01c267

Browse files
committed
Update for latest pytorch
1 parent bd3306f commit e01c267

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mmdet/core/utils/dist_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ def _allreduce_coalesced(tensors, world_size, bucket_size_mb=-1):
1414
else:
1515
buckets = OrderedDict()
1616
for tensor in tensors:
17-
tp = tensor.type()
17+
tp = tensor.scalar_type()
1818
if tp not in buckets:
1919
buckets[tp] = []
2020
buckets[tp].append(tensor)

0 commit comments

Comments
 (0)