()
| 168 | |
| 169 | |
| 170 | def parse_args(): |
| 171 | parser = argparse.ArgumentParser(description=__doc__) |
| 172 | parser.add_argument("--iters", type=int, default=20) |
| 173 | parser.add_argument("--warmup", type=int, default=5) |
| 174 | parser.add_argument( |
| 175 | "--configs", |
| 176 | type=str, |
| 177 | default=None, |
| 178 | help="Semicolon-separated 'tokens,hidden,vocab' tuples, " |
| 179 | "e.g. '4096,2048,32768;4096,2048,131072'.", |
| 180 | ) |
| 181 | args = parser.parse_args() |
| 182 | if args.configs: |
| 183 | args.configs = [tuple(int(x) for x in tup.split(",")) for tup in args.configs.split(";")] |
| 184 | else: |
| 185 | args.configs = DEFAULT_CONFIGS |
| 186 | return args |
| 187 | |
| 188 | |
| 189 | if __name__ == "__main__": |
no outgoing calls
no test coverage detected