[Fix]: Fix mae config name (#498)

pull/567/head
Yuan Liu 2022-10-05 13:45:08 +08:00 committed by Yixiao Fang
parent f9a85146fa
commit 088d4c8285
5 changed files with 5 additions and 5 deletions

View File

@ -1,4 +1,4 @@
_base_ = 'mae_vit-large-p16_8xb512-fp16-coslr-300e_in1k.py'
_base_ = 'mae_vit-large-p16_8xb512-amp-coslr-300e_in1k.py'
# pre-train for 100 epochs
train_cfg = dict(max_epochs=1600)

View File

@ -1,4 +1,4 @@
_base_ = 'mae_vit-base-p16_8xb512-fp16-coslr-1600e_in1k.py'
_base_ = 'mae_vit-base-p16_8xb512-amp-coslr-1600e_in1k.py'
# model settings
model = dict(

View File

@ -1,4 +1,4 @@
_base_ = 'mae_vit-base-p16_8xb512-fp16-coslr-300e_in1k.py'
_base_ = 'mae_vit-base-p16_8xb512-amp-coslr-300e_in1k.py'
# model settings
model = dict(

View File

@ -1,4 +1,4 @@
_base_ = 'mae_vit-base-p16_8xb512-fp16-coslr-400e_in1k.py'
_base_ = 'mae_vit-base-p16_8xb512-amp-coslr-400e_in1k.py'
# model settings
model = dict(

View File

@ -1,4 +1,4 @@
_base_ = 'mae_vit-base-p16_8xb512-fp16-coslr-800e_in1k.py'
_base_ = 'mae_vit-base-p16_8xb512-amp-coslr-800e_in1k.py'
# model settings
model = dict(