diff --git a/configs/benchmarks/classification/imagenet/swin-base_ft-8xb256-coslr-100e_in1k-224.py b/configs/benchmarks/classification/imagenet/swin-base_ft-8xb256-coslr-100e_in1k-224.py index 41b25e61d..6bb8e4432 100644 --- a/configs/benchmarks/classification/imagenet/swin-base_ft-8xb256-coslr-100e_in1k-224.py +++ b/configs/benchmarks/classification/imagenet/swin-base_ft-8xb256-coslr-100e_in1k-224.py @@ -26,10 +26,9 @@ dict(type='ToTensor'), dict(type='Normalize', **img_norm_cfg) ] - data = dict( samples_per_gpu=256, drop_last=False, - workers_per_gpu=8, + workers_per_gpu=32, train=dict(pipeline=train_pipeline), - val=dict(pipeline=test_pipeline)) + val=dict(pipeline=test_pipeline)) \ No newline at end of file