Pytorch implementation of MLPMixer
Project description
MLP Mixer Pytorch
Pytorch implementation of MLP-Mixer.
Sample usage
foo@bar:❯ pip install mlp_mixer
from mlp_mixer import MLPMixer
model = MLPMixer(
img_size=IMG_SZ,
img_channels=IMG_CHANNELS,
num_classes=NUM_CLASSES,
mixer_depth=DEPTH,
num_patches=NUM_PATCHES,
num_channels=NUM_CHANNELS,
expansion=EXPANSION,
dropout=DROPOUT,
)
Citations
@misc{tolstikhin2021mlpmixer,
title = {MLP-Mixer: An all-MLP Architecture for Vision},
author = {Ilya Tolstikhin and Neil Houlsby and Alexander Kolesnikov and Lucas Beyer and Xiaohua Zhai and Thomas Unterthiner and Jessica Yung and Daniel Keysers and Jakob Uszkoreit and Mario Lucic and Alexey Dosovitskiy},
year = {2021},
eprint = {2105.01601},
archivePrefix = {arXiv},
primaryClass = {cs.CV}
}
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
mlp-mixer-0.0.1.tar.gz
(3.8 kB
view hashes)
Built Distribution
Close
Hashes for mlp_mixer-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 18c7e5e5aed08557cf29563050a727944506b8b26b7ddc2ad430a1463bed499d |
|
MD5 | 2cbc7d4b678c5a1b7cbf267403ebdf98 |
|
BLAKE2b-256 | 4074fad92cf0c3c7661372d7c0ad1bfb5371c0e3b3aa32ec912c9ac94c1d35bb |