| 22 | |
| 23 | |
| 24 | def parse_args(): |
| 25 | parser = argparse.ArgumentParser() |
| 26 | parser.add_argument("-c", "--config", help="yaml configuration file", |
| 27 | default="config/inference.yaml") |
| 28 | parser.add_argument("--seed", help="random seed", type=int, default=0) |
| 29 | parser.add_argument("-o", "--output_dir", help="output directory", default="output") |
| 30 | parser.add_argument("-f", "--pdb_files", help="list of pdb files", nargs='*', default=[]) |
| 31 | args = parser.parse_known_args()[0] |
| 32 | args.output_dir = os.path.expanduser(args.output_dir) |
| 33 | args.output_dir = os.path.realpath(args.output_dir) |
| 34 | |
| 35 | return args |
| 36 | |
| 37 | |
| 38 | def set_seed(seed): |