diff --git a/configs/selfsup/_base_/datasets/imagenet_simmim.py b/configs/selfsup/_base_/datasets/imagenet_simmim.py index ddd5f866..df2681c2 100644 --- 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, 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 index 67522c93..4cd2a355 100644 --- 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)