修改参数。

2025-10-31 14:16:34 +08:00
parent bce02fc418
commit 9d8eccde84
5 changed files with 20 additions and 17 deletions
@@ -63,7 +63,8 @@ TASKS = {
        lr=8e-4,
        # triplet_margin=0.35,
        triplet_margin=0.5,
-        center_loss_weight=0.1,
+        # center_loss_weight=0.1,
+        center_loss_weight=0.7,
        aug_strength="medium",
    ),
    "processed_ingredient": TaskConfig(
@@ -820,6 +821,6 @@ if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("task", choices=list(TASKS.keys()), nargs="?", default="dish")
    args = parser.parse_args()
-    main("processed_ingredient")
-    # main("whole_ingredient")
+    # main("processed_ingredient")
+    main("whole_ingredient")
    # main("dish")