[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
|
# pre-train for 100 epochs
|
||||||
train_cfg = dict(max_epochs=1600)
|
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 settings
|
||||||
model = dict(
|
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 settings
|
||||||
model = dict(
|
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 settings
|
||||||
model = dict(
|
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 settings
|
||||||
model = dict(
|
model = dict(
|
||||||
|
|
Loading…
Reference in New Issue