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

About stochastic depth rate #13

Open JiaquanYe opened 2 years ago

JiaquanYe commented 2 years ago

It seems that the released code did not implement 'stochastic depth' in CoAtNet module, but it was mentioned in the appedix A.2 of paper.