()
| 89 | |
| 90 | |
| 91 | def main(): |
| 92 | parser = argparse.ArgumentParser( |
| 93 | description="Example LMDB creation" |
| 94 | ) |
| 95 | parser.add_argument("--output_file", type=str, default=None, |
| 96 | help="Path to write the database to", |
| 97 | required=True) |
| 98 | |
| 99 | args = parser.parse_args() |
| 100 | checksum = create_db(args.output_file) |
| 101 | |
| 102 | # For testing reading: |
| 103 | read_db_with_caffe2(args.output_file, checksum) |
| 104 | |
| 105 | |
| 106 | if __name__ == '__main__': |
no test coverage detected
searching dependent graphs…