/CoAtNet-tensorflow

This is a Tensorflow Keras implementation of CoAtNet

Primary LanguagePython

CoAtNet-tensorflow

Overview

This is a Tensorflow Keras implementation of CoAtNet refer from this repo.

img

Citation

@article{dai2021coatnet,
  title={CoAtNet: Marrying Convolution and Attention for All Data Sizes},
  author={Dai, Zihang and Liu, Hanxiao and Le, Quoc V and Tan, Mingxing},
  journal={arXiv preprint arXiv:2106.04803},
  year={2021}
}

REFERENCE

[1]https://arxiv.org/pdf/2106.04803.pdf