()
| 6 | |
| 7 | |
| 8 | def parse_args(): |
| 9 | parser = argparse.ArgumentParser(description='evaluate a model') |
| 10 | parser.add_argument('config', help='config file path', type=str) |
| 11 | parser.add_argument( |
| 12 | '--trainer_name', help='name for trainer', type=str, default=None) |
| 13 | parser.add_argument( |
| 14 | '--checkpoint_path', |
| 15 | help='checkpoint to be evaluated', |
| 16 | type=str, |
| 17 | default=None) |
| 18 | args = parser.parse_args() |
| 19 | return args |
| 20 | |
| 21 | |
| 22 | def main(): |
no outgoing calls
no test coverage detected
searching dependent graphs…