mirrors
/
mmselfsup
mirror of https://github.com/open-mmlab/mmselfsup.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Feature] Refine mae-h-448 fine-tuning config (#509) 

* [Feature]: Add ceph to vit-h-448

* [Feature]: Refine ViT-H 448 test setting

* [Fix]: Fix lr

* [Fix]: Use disk backend
pull/582/head
Yuan Liu 2022-10-14 16:40:53 +08:00 committed by Yixiao Fang
parent 8136a5c2cf
commit f8a33ae543
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
configs/benchmarks/classification/imagenet/vit-huge-p16_ft-32xb8-coslr-50e_in1k-448.py
View File

@ -37,7 +37,21 @@ train_pipeline = [
dict(type='PackClsInputs')
]
test_pipeline = [
dict(type='LoadImageFromFile', file_client_args=file_client_args),
dict(
type='ResizeEdge',
scale=512,
edge='short',
backend='pillow',
interpolation='bicubic'),
dict(type='CenterCrop', crop_size=448),
dict(type='PackClsInputs')
]
train_dataloader = dict(dataset=dict(pipeline=train_pipeline))
val_dataloader = dict(dataset=dict(pipeline=test_pipeline))
test_dataloader = dict(dataset=dict(pipeline=test_pipeline))
# learning rate settings
optim_wrapper = dict(optimizer=dict(lr=0.001, layer_decay_rate=0.75))
optim_wrapper = dict(optimizer=dict(lr=0.004, layer_decay_rate=0.75))