Unverified Commit 775dd2d8 authored by Emin Orhan's avatar Emin Orhan Committed by GitHub
Browse files

Fix use of `cutmix_alpha` in classification training references (#8448)

parent c585a515
...@@ -19,9 +19,9 @@ def get_mixup_cutmix(*, mixup_alpha, cutmix_alpha, num_classes, use_v2): ...@@ -19,9 +19,9 @@ def get_mixup_cutmix(*, mixup_alpha, cutmix_alpha, num_classes, use_v2):
) )
if cutmix_alpha > 0: if cutmix_alpha > 0:
mixup_cutmix.append( mixup_cutmix.append(
transforms_module.CutMix(alpha=mixup_alpha, num_classes=num_classes) transforms_module.CutMix(alpha=cutmix_alpha, num_classes=num_classes)
if use_v2 if use_v2
else RandomCutMix(num_classes=num_classes, p=1.0, alpha=mixup_alpha) else RandomCutMix(num_classes=num_classes, p=1.0, alpha=cutmix_alpha)
) )
if not mixup_cutmix: if not mixup_cutmix:
return None return None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment