simmim_swin-base-w6_16xb128-amp-coslr-100e_in1k-192px.py 132 Bytes
Newer Older
renzhc's avatar
renzhc committed
1
2
3
4
_base_ = 'simmim_swin-base-w6_8xb256-amp-coslr-100e_in1k-192px.py'

# dataset 16 GPUs x 128
train_dataloader = dict(batch_size=128)