From 367d0ca3c6f20f7150715b602ddd7f9f7831e20a Mon Sep 17 00:00:00 2001
From: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>
Date: Fri, 22 Jul 2022 11:47:41 +0800
Subject: [PATCH] [Fix] fix bugs of SimMIM config (#360)

---
 configs/selfsup/_base_/datasets/imagenet_simmim.py              | 2 +-
 .../simmim/simmim_swin-base_8xb256-coslr-100e_in1k-192.py       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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)