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

2022-07-22 11:47:41 +08:00 · 2022-07-22 11:47:41 +08:00 · 367d0ca3c6
parent 9c28733ddd
commit 367d0ca3c6
2 changed files with 2 additions and 2 deletions
--- a/configs/selfsup/_base_/datasets/imagenet_simmim.py
+++ b/configs/selfsup/_base_/datasets/imagenet_simmim.py
@ -20,7 +20,7 @@ if not prefetch:

 train_pipeline.append(
    dict(
-        type='BlockwiseMaskGenerator',
+        type='SimMIMMaskGenerator',
        input_size=192,
        mask_patch_size=32,
        model_patch_size=4,
--- 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)