mmpretrain/configs/conformer/conformer-small-p32_8xb128_...

9 lines
250 B
Python

_base_ = [
'../_base_/models/conformer/small-p32.py',
'../_base_/datasets/imagenet_bs64_swin_224.py',
'../_base_/schedules/imagenet_bs1024_adamw_conformer.py',
'../_base_/default_runtime.py'
]
train_dataloader = dict(batch_size=128)