Skip to content

GitLab

Unverified Commit c054a239 authored by Cao Yuhang's avatar Cao Yuhang Committed by GitHub
Browse files

remove assertion (#549)

parent 49e32c26
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
mmcv/parallel/collate.py
View file @ c054a239
...@@ -24,7 +24,6 @@ def collate(batch, samples_per_gpu=1): ...@@ -24,7 +24,6 @@ def collate(batch, samples_per_gpu=1):
raise TypeError(f'{batch.dtype} is not supported.') raise TypeError(f'{batch.dtype} is not supported.')
if isinstance(batch[0], DataContainer): if isinstance(batch[0], DataContainer):
assert len(batch) % samples_per_gpu == 0
stacked = [] stacked = []
if batch[0].cpu_only: if batch[0].cpu_only:
for i in range(0, len(batch), samples_per_gpu): for i in range(0, len(batch), samples_per_gpu):
......
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