fix imgs_per_gpu in 8GPU multihead linear evaluation

2020-07-19 18:11:57 +08:00 · 2020-07-19 18:11:57 +08:00 · 8d7094002f
parent 12a7c28573
commit 8d7094002f
2 changed files with 2 additions and 2 deletions
--- a/configs/benchmarks/linear_classification/imagenet/r50_multihead.py
+++ b/configs/benchmarks/linear_classification/imagenet/r50_multihead.py
@ -50,7 +50,7 @@ test_pipeline = [
    dict(type='Normalize', **img_norm_cfg),
 ]
 data = dict(
-    imgs_per_gpu=256,  # total 256
+    imgs_per_gpu=32,  # total 32x8=256
    workers_per_gpu=5,
    train=dict(
        type=dataset_type,
--- a/configs/benchmarks/linear_classification/imagenet/r50_multihead_sobel.py
+++ b/configs/benchmarks/linear_classification/imagenet/r50_multihead_sobel.py
@ -50,7 +50,7 @@ test_pipeline = [
    dict(type='Normalize', **img_norm_cfg),
 ]
 data = dict(
-    imgs_per_gpu=256,  # total 256
+    imgs_per_gpu=32,  # total 32x8=256
    workers_per_gpu=5,
    train=dict(
        type=dataset_type,