jaketae / mlp-mixer

PyTorch implementation of MLP-Mixer: An all-MLP Architecture for Vision

Hi there 👋 I'm Jake.

I'm passionate about generative modeling, text-to-speech, NLP, and recommendation systems.

I recently graduated from Yale University with a BSc in CS and Math. Previously, I was a

• Machine Learning & Software Engineer Intern at Facebook
• Machine Learning Engineer Intern at Hugging Face
• Machine Learning Research Intern at Neosapience

mlp-mixer's Issues

a small bug

hi, thanks for the great code! I was using it and find a small bug:

mlp-mixer/mlp_mixer/core.py

Line 59 in e7d68df

num_patches, num_features, expansion_factor, dropout

num_features, num_patches should be switched. same case for the channel_mixer.