()
| 10 | |
| 11 | |
| 12 | def get_parser(): |
| 13 | parser = argparse.ArgumentParser( |
| 14 | description="Instruction Induction.") |
| 15 | |
| 16 | parser.add_argument("--db_conf", type=str, |
| 17 | default = '../database/configs/config.ini') |
| 18 | |
| 19 | """ |
| 20 | parser.add_argument("--train_data", type=str, |
| 21 | default="./data/raw/train/rules.json") |
| 22 | parser.add_argument("--eval_data", type=str, |
| 23 | default="./data/raw/execute/zhenzhi.json") |
| 24 | |
| 25 | parser.add_argument("--data_save", type=str, |
| 26 | default="./result/{}/data/") |
| 27 | |
| 28 | parser.add_argument("--seed", type=int, default=42) |
| 29 | parser.add_argument("--runlog", type=str, |
| 30 | default="./result/{}/exp_runtime.log") |
| 31 | parser.add_argument("--logdir", type=str, |
| 32 | default="./result/{}/logdir/") |
| 33 | parser.add_argument("--model_save", type=str, |
| 34 | default="./result/{}/model/") |
| 35 | |
| 36 | parser.add_argument("--gen_sample", type=int, default=20) |
| 37 | parser.add_argument("--gen_demo", type=int, default=16) |
| 38 | parser.add_argument("--gen_prompt_per_sample", type=int, default=5) |
| 39 | parser.add_argument("--gen_model", type=str, default="text-davinci-003") |
| 40 | parser.add_argument("--gen_max_tokens", type=int, default=200) |
| 41 | |
| 42 | parser.add_argument("--eval_sample", type=int, default=20) |
| 43 | parser.add_argument("--eval_model", type=str, default="text-davinci-003") |
| 44 | parser.add_argument("--eval_max_tokens", type=int, default=1000) |
| 45 | |
| 46 | parser.add_argument("--storage_budget", type=int, default=500) # limit storage space of built indexes |
| 47 | """ |
| 48 | |
| 49 | |
| 50 | return parser |
| 51 | |
| 52 | |
| 53 | def set_logger(log_file): |
nothing calls this directly
no outgoing calls
no test coverage detected