Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
mlp-mixer-base-p16_64xb64_in1k.py 256 Bytes
Newer Older
renzhc's avatar
first commit  
renzhc committed
1
2
3
4
5
6
7
8 
_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',
]

optim_wrapper = dict(clip_grad=dict(max_norm=1.0))