- Notifications
You must be signed in to change notification settings - Fork 7.2k
Closed
Description
I believe the alpha argument here should be cutmix_alpha not mixup_alpha:
| transforms_module.CutMix(alpha=mixup_alpha, num_classes=num_classes) |
Same issue here as well:
| else RandomCutMix(num_classes=num_classes, p=1.0, alpha=mixup_alpha) |
Just curious: were the pretrained models actually trained using this code? Any chance this might have affected the performance of the pretrained models in any way?