()
| 126 | return table |
| 127 | |
| 128 | def main(): |
| 129 | parser = argparse.ArgumentParser(description="Print a concise summary of net computation.") |
| 130 | parser.add_argument('filename', help='net prototxt file to summarize') |
| 131 | parser.add_argument('-w', '--max-width', help='maximum field width', |
| 132 | type=int, default=30) |
| 133 | args = parser.parse_args() |
| 134 | |
| 135 | net = read_net(args.filename) |
| 136 | table = summarize_net(net) |
| 137 | print_table(table, max_width=args.max_width) |
| 138 | |
| 139 | if __name__ == '__main__': |
| 140 | main() |
no test coverage detected