search

microsoft / esvit

EsViT: Efficient self-supervised Vision Transformers
MIT License
408 stars 46 forks source link

Maybe a bug in SwinTrans #9

Closed BoPang1996 closed 3 years ago

BoPang1996 commented 3 years ago

https://github.com/microsoft/esvit/blob/c5d73eba76d76136a5ed162263b934df57ec04dc/models/swin_transformer.py#L300

In this line, should (self.H, self.W) be (H, W)?

ChunyuanLI commented 3 years ago

Thanks for pointing it out. This is corrected now.