1hunters / EdgeViT

This is an unofficial PyTorch implementation of EdgeViT in "EdgeViTs: Competing Light-weight CNNs on Mobile Devices with Vision Transformers", arXiv 2022.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

The model doesn't seem to fit

yanjingke opened this issue · comments

The model doesn't seem to fit

Hi,
I noticed that indeed the code missed an MLP in the LGL bottleneck, which has been added now. Thanks!