https://github.com/isaaccorley/convmlp-pytorch

PyTorch implementation of "ConvMLP: Hierarchical Convolutional MLPs for Vision" Li et al. (2021)
https://github.com/isaaccorley/convmlp-pytorch

Last synced: about 1 month ago
JSON representation

PyTorch implementation of "ConvMLP: Hierarchical Convolutional MLPs for Vision" Li et al. (2021)

• Host: GitHub
• URL: https://github.com/isaaccorley/convmlp-pytorch
• Owner: isaaccorley
• License: mit
• Created: 2021-09-12T00:19:58.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2021-09-12T02:25:16.000Z (over 3 years ago)
• Last Synced: 2024-10-14T10:15:32.368Z (3 months ago)
• Language: Python
• Size: 222 KB
• Stars: 4
• Watchers: 2
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# convmlp-pytorch

(WIP) PyTorch implementation of ["ConvMLP: Hierarchical Convolutional MLPs for Vision" Li et al. (2021)](https://arxiv.org/abs/2109.04454v1)



## Usage

```python

import torch

from convmlp import ConvMLP

model = ConvMLP(

        num_classes = 10,

        in_channels = 3,

        hidden_dim = 64,

        ratio = 1,

        levels = 3,

        c = 3,

        m = 2,

)

x = torch.randn(1, 3, 256, 256)

model(x) # (1, 10)

```