Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
SOLOv2-pytorch
Commits
3ef19ea7
Commit
3ef19ea7
authored
Oct 25, 2018
by
Kai Chen
Browse files
bug fix for init sampler during testing
parent
5f44ba8a
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
3 deletions
+4
-3
mmdet/datasets/loader/build_loader.py
mmdet/datasets/loader/build_loader.py
+4
-3
No files found.
mmdet/datasets/loader/build_loader.py
View file @
3ef19ea7
...
@@ -24,13 +24,14 @@ def build_dataloader(dataset,
...
@@ -24,13 +24,14 @@ def build_dataloader(dataset,
rank
)
rank
)
batch_size
=
imgs_per_gpu
batch_size
=
imgs_per_gpu
num_workers
=
workers_per_gpu
num_workers
=
workers_per_gpu
else
:
if
not
kwargs
.
get
(
'shuffle'
,
True
):
sampler
=
None
else
:
else
:
sampler
=
GroupSampler
(
dataset
,
imgs_per_gpu
)
sampler
=
GroupSampler
(
dataset
,
imgs_per_gpu
)
batch_size
=
num_gpus
*
imgs_per_gpu
batch_size
=
num_gpus
*
imgs_per_gpu
num_workers
=
num_gpus
*
workers_per_gpu
num_workers
=
num_gpus
*
workers_per_gpu
if
not
kwargs
.
get
(
'shuffle'
,
True
):
sampler
=
None
data_loader
=
DataLoader
(
data_loader
=
DataLoader
(
dataset
,
dataset
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment