mirror of
https://github.com/open-mmlab/mmselfsup.git
synced 2025-06-03 14:59:38 +08:00
5 lines
138 B
Python
5 lines
138 B
Python
_base_ = 'cae_vit-base-p16_16xb128-amp-coslr-300e_in1k.py'
|
|
|
|
# dataset 8GPUs x 256
|
|
train_dataloader = dict(batch_size=256, num_workers=16)
|