Skip to content

GitLab

Commit 5421859a authored by Kai Chen's avatar Kai Chen
Browse files

remove comments

parent db9aaac7
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 5 deletions
mmdet/core/parallel/scatter_gather.py
View file @ 5421859a
...@@ -16,7 +16,6 @@ def scatter(inputs, target_gpus, dim=0): ...@@ -16,7 +16,6 @@ def scatter(inputs, target_gpus, dim=0):
if isinstance(obj, torch.Tensor): if isinstance(obj, torch.Tensor):
return OrigScatter.apply(target_gpus, None, dim, obj) return OrigScatter.apply(target_gpus, None, dim, obj)
if isinstance(obj, DataContainer): if isinstance(obj, DataContainer):
# print('data container', obj)
if obj.cpu_only: if obj.cpu_only:
return obj.data return obj.data
else: else:
...@@ -24,14 +23,10 @@ def scatter(inputs, target_gpus, dim=0): ...@@ -24,14 +23,10 @@ def scatter(inputs, target_gpus, dim=0):
if isinstance(obj, tuple) and len(obj) > 0: if isinstance(obj, tuple) and len(obj) > 0:
return list(zip(*map(scatter_map, obj))) return list(zip(*map(scatter_map, obj)))
if isinstance(obj, list) and len(obj) > 0: if isinstance(obj, list) and len(obj) > 0:
# print('list', obj)
out = list(map(list, zip(*map(scatter_map, obj)))) out = list(map(list, zip(*map(scatter_map, obj))))
# print('list out', out)
return out return out
if isinstance(obj, dict) and len(obj) > 0: if isinstance(obj, dict) and len(obj) > 0:
# print('dict\n', obj)
out = list(map(type(obj), zip(*map(scatter_map, obj.items())))) out = list(map(type(obj), zip(*map(scatter_map, obj.items()))))
# print('dict output\n', out)
return out return out
return [obj for targets in target_gpus] return [obj for targets in target_gpus]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment