|
4 | 4 |
|
5 | 5 | def main():
|
6 | 6 | parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
|
7 |
| - parser.add_argument('--data_path', required=True, type=str, help="Path to the prepared data") |
8 |
| - parser.add_argument('--model_path', required=True, type=str, help="Path to where to save the model") |
9 |
| - parser.add_argument('--batch_size', nargs="?", type=int, default=100, help="Size of batch") |
10 |
| - parser.add_argument('--ensemble_count', nargs="?", type=int, default=8, help="Number of models to train") |
| 7 | + parser.add_argument('--data_path', required=True, type=str, help=" ") |
| 8 | + parser.add_argument('--model_path', required=True, type=str, help=" ") |
| 9 | + parser.add_argument('--batch_size', nargs="?", type=int, default=100, help=" ") |
| 10 | + parser.add_argument('--ensemble_count', nargs="?", type=int, default=8, help=" ") |
11 | 11 | parser.add_argument('--epoch_count', nargs="?", type=int, default=100, help=" ")
|
12 | 12 | parser.add_argument('--eval_frequency', nargs="?", type=int, default=4000, help=" ")
|
13 | 13 | parser.add_argument('--update_frequency', nargs="?", type=int, default=1, help=" ")
|
|
0 commit comments