ZhugeKongan / torch-template-for-deep-learning

Pytorch Implementations of large number classical backbone CNNs, data enhancement, torch loss, attention, visualization and some common algorithms.
Apache License 2.0
1.06k stars 191 forks source link

Should be `h=img.size(-2), w=img.size(-1)'? #1

Closed CiaoHe closed 2 years ago

CiaoHe commented 2 years ago

https://github.com/ZhugeKongan/torch-template-for-deep-learning/blob/e2a4ab70a00b45709da6ec5f1ae5c089b95b92da/autoaug/cutout.py#L22

ZhugeKongan commented 2 years ago

Thank you for your reminder, it has been corrected.