diff --git a/configs/selfsup/byol/byol_resnet50_8xb256-fp16-accum2-coslr-100e_in1k.py b/configs/selfsup/byol/byol_resnet50_8xb256-amp-accum2-coslr-100e_in1k.py similarity index 87% rename from configs/selfsup/byol/byol_resnet50_8xb256-fp16-accum2-coslr-100e_in1k.py rename to configs/selfsup/byol/byol_resnet50_8xb256-amp-accum2-coslr-100e_in1k.py index 9ed9ae8b..c904e927 100644 --- a/configs/selfsup/byol/byol_resnet50_8xb256-fp16-accum2-coslr-100e_in1k.py +++ b/configs/selfsup/byol/byol_resnet50_8xb256-amp-accum2-coslr-100e_in1k.py @@ -1,4 +1,4 @@ -_base_ = 'byol_resnet50_8xb256-fp16-accum2-coslr-200e_in1k.py' +_base_ = 'byol_resnet50_8xb256-amp-accum2-coslr-200e_in1k.py' # optimizer optimizer = dict(lr=7.2) diff --git a/configs/selfsup/byol/byol_resnet50_8xb256-fp16-accum2-coslr-200e_in1k.py b/configs/selfsup/byol/byol_resnet50_8xb256-amp-accum2-coslr-200e_in1k.py similarity index 100% rename from configs/selfsup/byol/byol_resnet50_8xb256-fp16-accum2-coslr-200e_in1k.py rename to configs/selfsup/byol/byol_resnet50_8xb256-amp-accum2-coslr-200e_in1k.py diff --git a/configs/selfsup/byol/byol_resnet50_8xb256-fp16-accum2-coslr-300e_in1k.py b/configs/selfsup/byol/byol_resnet50_8xb256-amp-accum2-coslr-300e_in1k.py similarity index 85% rename from configs/selfsup/byol/byol_resnet50_8xb256-fp16-accum2-coslr-300e_in1k.py rename to configs/selfsup/byol/byol_resnet50_8xb256-amp-accum2-coslr-300e_in1k.py index 77c61d20..2c58ad32 100644 --- a/configs/selfsup/byol/byol_resnet50_8xb256-fp16-accum2-coslr-300e_in1k.py +++ b/configs/selfsup/byol/byol_resnet50_8xb256-amp-accum2-coslr-300e_in1k.py @@ -1,4 +1,4 @@ -_base_ = 'byol_resnet50_8xb256-fp16-accum2-coslr-200e_in1k.py' +_base_ = 'byol_resnet50_8xb256-amp-accum2-coslr-200e_in1k.py' # learning rate scheduler param_scheduler = [ 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-amp-coslr-300e_in1k.py similarity index 56% rename from configs/selfsup/cae/cae_vit-base-p16_16xb128-fp16-coslr-300e_in1k.py rename to configs/selfsup/cae/cae_vit-base-p16_16xb128-amp-coslr-300e_in1k.py index 6eb31538..43b60daa 100644 --- a/configs/selfsup/cae/cae_vit-base-p16_16xb128-fp16-coslr-300e_in1k.py +++ b/configs/selfsup/cae/cae_vit-base-p16_16xb128-amp-coslr-300e_in1k.py @@ -1,4 +1,4 @@ -_base_ = 'cae_vit-base-p16_32xb64-fp16-coslr-300e_in1k.py' +_base_ = 'cae_vit-base-p16_32xb64-amp-coslr-300e_in1k.py' # dataset 128 x 16 train_dataloader = dict(batch_size=128, num_workers=16) diff --git a/configs/selfsup/cae/cae_vit-base-p16_32xb64-fp16-coslr-300e_in1k.py b/configs/selfsup/cae/cae_vit-base-p16_32xb64-amp-coslr-300e_in1k.py similarity index 100% rename from configs/selfsup/cae/cae_vit-base-p16_32xb64-fp16-coslr-300e_in1k.py rename to configs/selfsup/cae/cae_vit-base-p16_32xb64-amp-coslr-300e_in1k.py 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-amp-coslr-300e_in1k.py similarity index 56% rename from configs/selfsup/cae/cae_vit-base-p16_8xb256-fp16-coslr-300e_in1k.py rename to configs/selfsup/cae/cae_vit-base-p16_8xb256-amp-coslr-300e_in1k.py index 596d2c2e..49d2896b 100644 --- a/configs/selfsup/cae/cae_vit-base-p16_8xb256-fp16-coslr-300e_in1k.py +++ b/configs/selfsup/cae/cae_vit-base-p16_8xb256-amp-coslr-300e_in1k.py @@ -1,4 +1,4 @@ -_base_ = 'cae_vit-base-p16_16xb128-fp16-coslr-300e_in1k.py' +_base_ = 'cae_vit-base-p16_16xb128-amp-coslr-300e_in1k.py' # dataset 8GPUs x 256 train_dataloader = dict(batch_size=256, num_workers=16) diff --git a/configs/selfsup/mae/mae_vit-base-p16_8xb512-fp16-coslr-1600e_in1k.py b/configs/selfsup/mae/mae_vit-base-p16_8xb512-amp-coslr-1600e_in1k.py similarity index 87% rename from configs/selfsup/mae/mae_vit-base-p16_8xb512-fp16-coslr-1600e_in1k.py rename to configs/selfsup/mae/mae_vit-base-p16_8xb512-amp-coslr-1600e_in1k.py index 6ad60dd8..dad6bdac 100644 --- a/configs/selfsup/mae/mae_vit-base-p16_8xb512-fp16-coslr-1600e_in1k.py +++ b/configs/selfsup/mae/mae_vit-base-p16_8xb512-amp-coslr-1600e_in1k.py @@ -1,4 +1,4 @@ -_base_ = 'mae_vit-base-p16_8xb512-coslr-400e-fp16_in1k.py' +_base_ = 'mae_vit-base-p16_8xb512-amp-coslr-400e_in1k.py' # pre-train for 800 epochs train_cfg = dict(max_epochs=1600) diff --git a/configs/selfsup/mae/mae_vit-base-p16_8xb512-fp16-coslr-300e_in1k.py b/configs/selfsup/mae/mae_vit-base-p16_8xb512-amp-coslr-300e_in1k.py similarity index 87% rename from configs/selfsup/mae/mae_vit-base-p16_8xb512-fp16-coslr-300e_in1k.py rename to configs/selfsup/mae/mae_vit-base-p16_8xb512-amp-coslr-300e_in1k.py index 61171e73..21414b83 100644 --- a/configs/selfsup/mae/mae_vit-base-p16_8xb512-fp16-coslr-300e_in1k.py +++ b/configs/selfsup/mae/mae_vit-base-p16_8xb512-amp-coslr-300e_in1k.py @@ -1,4 +1,4 @@ -_base_ = 'mae_vit-base-p16_8xb512-coslr-400e-fp16_in1k.py' +_base_ = 'mae_vit-base-p16_8xb512-amp-coslr-400e_in1k.py' # pre-train for 800 epochs train_cfg = dict(max_epochs=300) diff --git a/configs/selfsup/mae/mae_vit-base-p16_8xb512-fp16-coslr-400e_in1k.py b/configs/selfsup/mae/mae_vit-base-p16_8xb512-amp-coslr-400e_in1k.py similarity index 100% rename from configs/selfsup/mae/mae_vit-base-p16_8xb512-fp16-coslr-400e_in1k.py rename to configs/selfsup/mae/mae_vit-base-p16_8xb512-amp-coslr-400e_in1k.py diff --git a/configs/selfsup/mae/mae_vit-base-p16_8xb512-fp16-coslr-800e_in1k.py b/configs/selfsup/mae/mae_vit-base-p16_8xb512-amp-coslr-800e_in1k.py similarity index 87% rename from configs/selfsup/mae/mae_vit-base-p16_8xb512-fp16-coslr-800e_in1k.py rename to configs/selfsup/mae/mae_vit-base-p16_8xb512-amp-coslr-800e_in1k.py index 6563ce01..218a93b5 100644 --- a/configs/selfsup/mae/mae_vit-base-p16_8xb512-fp16-coslr-800e_in1k.py +++ b/configs/selfsup/mae/mae_vit-base-p16_8xb512-amp-coslr-800e_in1k.py @@ -1,4 +1,4 @@ -_base_ = 'mae_vit-base-p16_8xb512-coslr-400e-fp16_in1k.py' +_base_ = 'mae_vit-base-p16_8xb512-amp-coslr-400e_in1k.py' # pre-train for 800 epochs train_cfg = dict(max_epochs=800) diff --git a/configs/selfsup/mocov3/mocov3_vit-small-p16_16xb256-fp16-coslr-300e_in1k-224.py b/configs/selfsup/mocov3/mocov3_vit-small-p16_16xb256-amp-coslr-300e_in1k.py similarity index 100% rename from configs/selfsup/mocov3/mocov3_vit-small-p16_16xb256-fp16-coslr-300e_in1k-224.py rename to configs/selfsup/mocov3/mocov3_vit-small-p16_16xb256-amp-coslr-300e_in1k.py diff --git a/configs/selfsup/rotation_pred/rotation-pred_resnet50_8xb16-fp16-steplr-70e_in1k.py b/configs/selfsup/rotation_pred/rotation-pred_resnet50_8xb16-amp-steplr-70e_in1k.py similarity index 100% rename from configs/selfsup/rotation_pred/rotation-pred_resnet50_8xb16-fp16-steplr-70e_in1k.py rename to configs/selfsup/rotation_pred/rotation-pred_resnet50_8xb16-amp-steplr-70e_in1k.py diff --git a/configs/selfsup/simmim/simmim_swin-base_16xb128-coslr-100e_in1k-192.py b/configs/selfsup/simmim/simmim_swin-base_16xb128-amp-coslr-100e_in1k-192.py similarity index 100% rename from configs/selfsup/simmim/simmim_swin-base_16xb128-coslr-100e_in1k-192.py rename to configs/selfsup/simmim/simmim_swin-base_16xb128-amp-coslr-100e_in1k-192.py diff --git a/configs/selfsup/simmim/simmim_swin-base_8xb256-coslr-100e_in1k-192.py b/configs/selfsup/simmim/simmim_swin-base_8xb256-amp-coslr-100e_in1k-192.py similarity index 55% rename from configs/selfsup/simmim/simmim_swin-base_8xb256-coslr-100e_in1k-192.py rename to configs/selfsup/simmim/simmim_swin-base_8xb256-amp-coslr-100e_in1k-192.py index 23b14b8d..acaf0594 100644 --- a/configs/selfsup/simmim/simmim_swin-base_8xb256-coslr-100e_in1k-192.py +++ b/configs/selfsup/simmim/simmim_swin-base_8xb256-amp-coslr-100e_in1k-192.py @@ -1,4 +1,4 @@ -_base_ = 'simmim_swin-base_16xb128-coslr-100e_in1k-192.py' +_base_ = 'simmim_swin-base_16xb128-amp-coslr-100e_in1k-192.py' # dataset 8 GPUs x 256 train_dataloader = dict(batch_size=256, num_workers=16)