search

vztu / maxim-pytorch

[CVPR 2022 Oral] PyTorch re-implementation for "MAXIM: Multi-Axis MLP for Image Processing", with *training code*. Official Jax repo: https://github.com/google-research/maxim
Apache License 2.0
166 stars 22 forks source link

Excuse me, are the GridGatingUnit and other modules not completed? #7

Open zhoujun501 opened 1 year ago

zhoujun501 commented 1 year ago

Excuse me, are the GridGatingUnit and other modules not completed?

vztu commented 1 year ago

Please check out our clean implementation here: https://github.com/vztu/maxim-pytorch/tree/main/maxim_pytorch.