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

eminorhan · web-flow · commit 775dd2d8f8e1 · 2024-05-29T12:58:33.000+01:00
diff --git a/references/classification/transforms.py b/references/classification/transforms.py
@@ -19,9 +19,9 @@ def get_mixup_cutmix(*, mixup_alpha, cutmix_alpha, num_classes, use_v2):
         )
     if cutmix_alpha > 0:
         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
-            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:
         return None

Original file line number	Diff line number	Diff line change
`@@ -19,9 +19,9 @@ def get_mixup_cutmix(*, mixup_alpha, cutmix_alpha, num_classes, use_v2):`
`19`	`19`	`)`
`20`	`20`	`if cutmix_alpha > 0:`
`21`	`21`	`mixup_cutmix.append(`
`22`		`- transforms_module.CutMix(alpha=mixup_alpha, num_classes=num_classes)`
	`22`	`+ transforms_module.CutMix(alpha=cutmix_alpha, num_classes=num_classes)`
`23`	`23`	`if use_v2`
`24`		`- else RandomCutMix(num_classes=num_classes, p=1.0, alpha=mixup_alpha)`
	`24`	`+ else RandomCutMix(num_classes=num_classes, p=1.0, alpha=cutmix_alpha)`
`25`	`25`	`)`
`26`	`26`	`if not mixup_cutmix:`
`27`	`27`	`return None`