| 28 | |
| 29 | |
| 30 | def parse_args(): |
| 31 | parser = argparse.ArgumentParser() |
| 32 | parser.add_argument("--client_config", type=str) |
| 33 | parser.add_argument("--connect", type=str) |
| 34 | parser.add_argument("--use_gpu", type=str) |
| 35 | parser.add_argument("--data_dir", type=str) |
| 36 | parser.add_argument("--reader_file", type=str) |
| 37 | parser.add_argument("--batchsize", type=int) |
| 38 | parser.add_argument("--client_mode", type=str) |
| 39 | args = parser.parse_args() |
| 40 | args.use_gpu = (True if args.use_gpu.lower() == "true" else False) |
| 41 | return args |
| 42 | |
| 43 | |
| 44 | def create_data_loader(args): |