chinhsuanwu / coatnet-pytorch

A PyTorch implementation of "CoAtNet: Marrying Convolution and Attention for All Data Sizes"
https://arxiv.org/abs/2106.04803
MIT License
370 stars 67 forks source link

Hello, first off, really appreciate your work! #8

Open learningelectric opened 2 years ago

learningelectric commented 2 years ago

Hello, first off, really appreciate your work! Unfortunately, I'm getting overfitting using a custom dataset even in coatnet_0, is there a workaround? Please! Thank you veru much!