mmpretrain/configs/conformer/conformer-base-p16_8xb128_in1k.py
2022-07-18 11:11:13 +08:00

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)