MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / main

Function main

samples/python/detectron2/build_engine.py:216–220  ·  view source on GitHub ↗
(args)

Source from the content-addressed store, hash-verified

214
215
216def main(args):
217 builder = EngineBuilder(args.verbose, args.workspace)
218 builder.create_network(args.onnx)
219 builder.create_engine(args.engine, args.precision, args.det2_config, args.calib_input, args.calib_cache, args.calib_num_images,
220 args.calib_batch_size)
221
222
223if __name__ == "__main__":

Callers 1

build_engine.pyFile · 0.70

Calls 3

create_networkMethod · 0.95
create_engineMethod · 0.95
EngineBuilderClass · 0.70

Tested by

no test coverage detected