mmpretrain/configs/conformer/conformer-base-p16_8xb128_i...

9 lines
249 B
Python

_base_ = [
'../_base_/models/conformer/base-p16.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)