/res-mlp

PyTorch implementation of ResMLP: Feedforward networks for image classification with data-efficient training

Primary LanguagePythonMIT LicenseMIT

ResMLP

PyTorch implementation of ResMLP: Feedforward networks for image classification with data-efficient training.

Quickstart

Clone this repository.

git clone https://github.com/jaketae/res-mlp.git

Navigate to the cloned directory. You can start using the model via

>>> from res_mlp import ResMLP
>>> model = ResMLP()

By default, the model comes with the following parameters:

ResMLP(
    image_size=256,
    patch_size=16,
    in_channels=3,
    num_features=128,
    expansion_factor=2,
    num_layers=6,
    num_classes=10,
)

Resources