()
| 3 | |
| 4 | |
| 5 | def parse_args(): |
| 6 | parser = argparse.ArgumentParser() |
| 7 | parser.add_argument('checkpoint', help='checkpoint file') |
| 8 | parser.add_argument( |
| 9 | '--save-path', type=str, required=True, help='destination file name') |
| 10 | args = parser.parse_args() |
| 11 | return args |
| 12 | |
| 13 | |
| 14 | def main(): |