[Fix]: Fix name bug

open-mmlab · Feb 25, 2022 · 4084b8c · 4084b8c
1 parent 910b8ce
commit 4084b8c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/configs/benchmarks/classification/imagenet/vit-b-p16_ft-8xb128-coslr-100e_in1k.py b/configs/benchmarks/classification/imagenet/vit-b-p16_ft-8xb128-coslr-100e_in1k.py
@@ -27,7 +27,7 @@
     dict(type='Normalize', **img_norm_cfg)
 ]
 data = dict(
-    imgs_per_gpu=128,
+    samples_per_gpu=128,
     drop_last=False,
     workers_per_gpu=32,
     train=dict(pipeline=train_pipeline),

diff --git a/configs/selfsup/mae/mae_vit-base-p16_8xb512-coslr-400e_in1k.py b/configs/selfsup/mae/mae_vit-base-p16_8xb512-coslr-400e_in1k.py
@@ -6,7 +6,7 @@
 ]
 
 # dataset
-data = dict(imgs_per_gpu=512, workers_per_gpu=32)
+data = dict(samples_per_gpu=512, workers_per_gpu=32)
 
 # optimizer
 optimizer = dict(