SHI-Labs / Compact-Transformers

Escaping the Big Data Paradigm with Compact Transformers, 2021 (Train your Vision Transformers in 30 mins on CIFAR-10 with a single GPU!)
https://arxiv.org/abs/2104.05704
Apache License 2.0
495 stars 77 forks source link

Mask forward fix #51

Closed alihassanijr closed 2 years ago

alihassanijr commented 2 years ago

Forwarded mask was not returned by mistake.