|
_base_ = [
|
|
'../_base_/models/mlp_mixer_base_patch16.py',
|
|
'../_base_/datasets/imagenet_bs64_mixer_224.py',
|
|
'../_base_/schedules/imagenet_bs4096_AdamW.py',
|
|
'../_base_/default_runtime.py',
|
|
]
|
|
|
|
default_hooks = dict(optimizer=dict(grad_clip=dict(max_norm=1.0)))
|