[Fix]: Fix mae config name (#498)
parent
f9a85146fa
commit
088d4c8285
|
@ -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)
|
||||
|
|
|
@ -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(
|
||||
|
|
|
@ -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(
|
||||
|
|
|
@ -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(
|
||||
|
|
|
@ -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(
|
||||
|
|
Loading…
Reference in New Issue