Skip to content

GitLab

Unverified Commit c8c34938 authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

Merge pull request #33 from OneDirection9/master 

bug fixes
parents 8a3e232c 2882ef12
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmcv/parallel/collate.py
View file @ c8c34938
...@@ -36,7 +36,7 @@ def collate(batch, samples_per_gpu=1): ...@@ -36,7 +36,7 @@ def collate(batch, samples_per_gpu=1):
assert isinstance(batch[i].data, torch.Tensor) assert isinstance(batch[i].data, torch.Tensor)
# TODO: handle tensors other than 3d # TODO: handle tensors other than 3d
assert batch[i].dim() == 3 assert batch[i].dim() == 3
c, h, w = batch[0].size() c, h, w = batch[i].size()
for sample in batch[i:i + samples_per_gpu]: for sample in batch[i:i + samples_per_gpu]:
assert c == sample.size(0) assert c == sample.size(0)
h = max(h, sample.size(1)) h = max(h, sample.size(1))
......
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