debug

PaddlePaddle · Jun 29, 2023 · 20d0786 · 20d0786
1 parent 69cbcab
commit 20d0786
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tasks/ssl/cae/main_finetune.py b/tasks/ssl/cae/main_finetune.py
@@ -455,13 +455,13 @@ def main(args):
         "alpha": args.mixup,
         "prob": args.mixup_switch_prob,
         "epsilon": args.smoothing,
-        "num_classes": args.nb_classes
+        "class_num": args.nb_classes
     }
     batch_transform_ops['Cutmix'] = {
         "alpha": args.cutmix,
         "prob": args.mixup_switch_prob,
         "epsilon": args.smoothing,
-        "num_classes": args.nb_classes
+        "class_num": args.nb_classes
     }
     mixup_fn = transforms.TransformOpSampler(**batch_transform_ops)
 

diff --git a/tasks/ssl/mae/main_finetune.py b/tasks/ssl/mae/main_finetune.py
@@ -317,13 +317,13 @@ def main(args):
         "alpha": args.mixup,
         "prob": args.mixup_switch_prob,
         "epsilon": args.smoothing,
-        "num_classes": args.nb_classes
+        "class_num": args.nb_classes
     }
     batch_transform_ops['Cutmix'] = {
         "alpha": args.cutmix,
         "prob": args.mixup_switch_prob,
         "epsilon": args.smoothing,
-        "num_classes": args.nb_classes
+        "class_num": args.nb_classes
     }
     mixup_fn = transforms.TransformOpSampler(**batch_transform_ops)