()
| 32 | |
| 33 | |
| 34 | def parse_option(): |
| 35 | parser = argparse.ArgumentParser( |
| 36 | 'TinyViT saving sparse logits script', add_help=False) |
| 37 | add_common_args(parser) |
| 38 | parser.add_argument('--check-saved-logits', |
| 39 | action='store_true', help='Check saved logits') |
| 40 | parser.add_argument('--skip-eval', |
| 41 | action='store_true', help='Skip evaluation') |
| 42 | |
| 43 | args = parser.parse_args() |
| 44 | |
| 45 | config = get_config(args) |
| 46 | |
| 47 | return args, config |
| 48 | |
| 49 | |
| 50 | def main(config): |
no test coverage detected