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

Method __init__

quickstart/IntroNotebooks/onnx_helper.py:28–34  ·  view source on GitHub ↗
(self, file, num_classes, target_dtype = np.float32)

Source from the content-addressed store, hash-verified

26
27class ONNXClassifierWrapper():
28 def __init__(self, file, num_classes, target_dtype = np.float32):
29
30 self.target_dtype = target_dtype
31 self.num_classes = num_classes
32 self.load(file)
33
34 self.stream = None
35
36 def load(self, file):
37 f = open(file, "rb")

Callers

nothing calls this directly

Calls 1

loadMethod · 0.95

Tested by

no test coverage detected