mirrors
/
deit
mirror of https://github.com/facebookresearch/deit
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

BUG Fix: ensure the right loss function is selected when mixtup or cutmix is enabeled

pull/91/head
Maxwell Aladago 2021-05-02 10:27:25 -04:00
parent 686507a6a5
commit 2f76ec64fd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@ -310,7 +310,7 @@ def main(args):
criterion = LabelSmoothingCrossEntropy() criterion = LabelSmoothingCrossEntropy()
if mixup_active > 0.: if mixup_active:
# smoothing is handled with mixup label transform # smoothing is handled with mixup label transform
criterion = SoftTargetCrossEntropy() criterion = SoftTargetCrossEntropy()
elif args.smoothing: elif args.smoothing: