mlp-mixer
mlp-mixer copied to clipboard
Published
20 hours ago •
jaketae
jaketae
PyTorch implementation of MLP-Mixer: An all-MLP Architecture for Vision
Results
2
mlp-mixer issues
Sort by
recently updated
recently updated
newest added
hi, thanks for the great code! I was using it and find a small bug: https://github.com/jaketae/mlp-mixer/blob/e7d68dfc31e94721724689e6ec90f05806b50124/mlp_mixer/core.py#L59 num_features, num_patches should be switched. same case for the channel_mixer.
First of all, thank you for your great creation. Is there a 3D data version of MLP-Mixer
