mirror of
https://github.com/open-mmlab/mmselfsup.git
synced 2025-06-03 14:59:38 +08:00
5 lines
140 B
Python
5 lines
140 B
Python
_base_ = 'mae_vit-base-p16_8xb512-coslr-400e_in1k.py'
|
|
|
|
# mixed precision
|
|
optim_wrapper = dict(type='AmpOptimWrapper', loss_scale='dynamic')
|