[Fix] fix bugs of SimMIM config (open-mmlab#360)

Betciso · Jul 28, 2022 · 367d0ca · 367d0ca
1 parent 9c28733
commit 367d0ca
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/configs/selfsup/_base_/datasets/imagenet_simmim.py b/configs/selfsup/_base_/datasets/imagenet_simmim.py
@@ -20,7 +20,7 @@
 
 train_pipeline.append(
     dict(
-        type='BlockwiseMaskGenerator',
+        type='SimMIMMaskGenerator',
         input_size=192,
         mask_patch_size=32,
         model_patch_size=4,

diff --git a/configs/selfsup/simmim/simmim_swin-base_8xb256-coslr-100e_in1k-192.py b/configs/selfsup/simmim/simmim_swin-base_8xb256-coslr-100e_in1k-192.py
@@ -1,4 +1,4 @@
-_base_ = 'simmim_swin-base_16xb128-coslr-100e_in1k.py'
+_base_ = 'simmim_swin-base_16xb128-coslr-100e_in1k-192.py'
 
 # data
 data = dict(samples_per_gpu=256)