/coatnet-pytorch

A PyTorch implementation of "CoAtNet: Marrying Convolution and Attention for All Data Sizes".

Primary LanguagePythonMIT LicenseMIT

Stargazers