更改配置文件

2025-11-11 09:02:45 +08:00
parent f204799b3e
commit 12e231dc6c
5 changed files with 20 additions and 20 deletions
@@ -347,8 +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='whole_ingredient')
+    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)