修改配置。

2025-11-07 09:40:54 +08:00
parent 3ca5d6e7d0
commit 45ed6b905a
5 changed files with 20 additions and 19 deletions
@@ -347,7 +347,8 @@ def main(task_key: str = 'dish', s: float = 64.0, m: float = 0.35, num_epochs: i
 if __name__ == '__main__':
    import argparse
    parser = argparse.ArgumentParser()
-    parser.add_argument('task', choices=list(TASKS.keys()), nargs='?', default='dish')
+    # parser.add_argument('task', choices=list(TASKS.keys()), nargs='?', default='dish')
+    parser.add_argument('task', choices=list(TASKS.keys()), nargs='?', default='whole_ingredient')
    parser.add_argument('--s', type=float, default=64.0)
    parser.add_argument('--m', type=float, default=0.35)
    parser.add_argument('--epochs', type=int, default=60)