(args)
| 264 | |
| 265 | |
| 266 | def main(args): |
| 267 | builder = EngineBuilder(args.verbose, args.workspace) |
| 268 | builder.create_network(args.onnx, args.batch_size, args.dynamic_batch_size) |
| 269 | if args.precision == "mixed": |
| 270 | builder.set_mixed_precision() |
| 271 | builder.create_engine(args.engine, args.precision, args.calib_input, args.calib_cache, args.calib_num_images, |
| 272 | args.calib_batch_size) |
| 273 | |
| 274 | |
| 275 | if __name__ == "__main__": |
no test coverage detected