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)