Skip to content

Commit

Permalink
[fbsync] Fix use of cutmix_alpha in classification training referen…
Browse files Browse the repository at this point in the history
…ces (#8448)

Reviewed By: vmoens

Differential Revision: D58283861

fbshipit-source-id: e7bbe19cf3298011373a5105227342a850acc868
  • Loading branch information
NicolasHug authored and facebook-github-bot committed Jun 7, 2024
1 parent 0562539 commit f3f318c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions references/classification/transforms.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit f3f318c

Please sign in to comment.