From bbcc8cbbb7d8b9d655a171d14d2d7f3002551950 Mon Sep 17 00:00:00 2001
From: YuanLiuuuuuu <3463423099@qq.com>
Date: Fri, 22 Apr 2022 21:47:01 +0800
Subject: [PATCH] [Fix]: cae pre-train config bug

---
 .../cae/cae_vit-base-p16_16xb128-fp16-coslr-300e_in1k.py        | 2 +-
 .../selfsup/cae/cae_vit-base-p16_8xb256-fp16-coslr-300e_in1k.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/configs/selfsup/cae/cae_vit-base-p16_16xb128-fp16-coslr-300e_in1k.py b/configs/selfsup/cae/cae_vit-base-p16_16xb128-fp16-coslr-300e_in1k.py
index bbf1dea39..e460c0eef 100644
--- a/configs/selfsup/cae/cae_vit-base-p16_16xb128-fp16-coslr-300e_in1k.py
+++ b/configs/selfsup/cae/cae_vit-base-p16_16xb128-fp16-coslr-300e_in1k.py
@@ -1,4 +1,4 @@
-_base_ = 'cae_vit-base-p16_32xb64-coslr-300e_in1k.py'
+_base_ = 'cae_vit-base-p16_32xb64-fp16-coslr-300e_in1k.py'
 
 # dataset
 data = dict(samples_per_gpu=128, workers_per_gpu=8)
diff --git a/configs/selfsup/cae/cae_vit-base-p16_8xb256-fp16-coslr-300e_in1k.py b/configs/selfsup/cae/cae_vit-base-p16_8xb256-fp16-coslr-300e_in1k.py
index b562644bd..f98b546a6 100644
--- a/configs/selfsup/cae/cae_vit-base-p16_8xb256-fp16-coslr-300e_in1k.py
+++ b/configs/selfsup/cae/cae_vit-base-p16_8xb256-fp16-coslr-300e_in1k.py
@@ -1,4 +1,4 @@
-_base_ = 'cae_vit-base-p16_16xb128-coslr-300e_in1k.py'
+_base_ = 'cae_vit-base-p16_16xb128-fp16-coslr-300e_in1k.py'
 
 # dataset
 data = dict(samples_per_gpu=256, workers_per_gpu=8)