From 8d7094002ffea18108195e94a1509eb493e43384 Mon Sep 17 00:00:00 2001 From: xiaohangzhan Date: Sun, 19 Jul 2020 18:11:57 +0800 Subject: [PATCH] fix imgs_per_gpu in 8GPU multihead linear evaluation --- .../benchmarks/linear_classification/imagenet/r50_multihead.py | 2 +- .../linear_classification/imagenet/r50_multihead_sobel.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/configs/benchmarks/linear_classification/imagenet/r50_multihead.py b/configs/benchmarks/linear_classification/imagenet/r50_multihead.py index 5cbe5f2e..94d3c091 100644 --- 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, diff --git a/configs/benchmarks/linear_classification/imagenet/r50_multihead_sobel.py b/configs/benchmarks/linear_classification/imagenet/r50_multihead_sobel.py index 2960c50e..98b92345 100644 --- 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,