mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
mmpretrain/configs/mlp_mixer/mlp-mixer-base-p16_64xb64_i...

9 lines
272 B
Python
Raw Normal View History

[Feature] Add MLP Mixer Backbone. (#528) * Add Mlp Mixer * Fix token_mix FNN add_identity and add out_indices * Fix mlp_mixer unit test
2021-11-15 18:25:22 +08:00
_base_ = [
'../_base_/models/mlp_mixer_base_patch16.py',
[Enhance] Add metafile, readme and converted models for MLP-Mixer (#539) * add pth converter * minor update on config files, add metafile and readme * add missing readme and minor fixes * minor fixes * Update config names and checkpoint download link * Update model_zoo.md Co-authored-by: mzr1996 <mzr1996@163.com>
2021-11-24 19:04:19 +08:00
'../_base_/datasets/imagenet_bs64_mixer_224.py',
[Feature] Add MLP Mixer Backbone. (#528) * Add Mlp Mixer * Fix token_mix FNN add_identity and add out_indices * Fix mlp_mixer unit test
2021-11-15 18:25:22 +08:00
'../_base_/schedules/imagenet_bs4096_AdamW.py',
'../_base_/default_runtime.py',
]
Refactor default hooks configs
2022-05-23 16:43:23 +08:00
default_hooks = dict(optimizer=dict(grad_clip=dict(max_norm=1.0)))