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

Method __init__

demo/HuggingFace/NNDF/models.py:497–505  ·  view source on GitHub ↗
(
        self,
        model: str,
        default_converter: ModelFileConverter = None,
        network_metadata: NetworkMetadata = None,
    )

Source from the content-addressed store, hash-verified

495 return network_definition
496
497 def __init__(
498 self,
499 model: str,
500 default_converter: ModelFileConverter = None,
501 network_metadata: NetworkMetadata = None,
502 ):
503 super().__init__(default_converter, network_metadata)
504 self.fpath = model
505 self.max_trt_workspace = 3072
506
507 def cleanup(self) -> None:
508 G_LOGGER.debug("Removing saved engine model from location: {}".format(self.fpath))

Callers

nothing calls this directly

Calls 1

__init__Method · 0.45

Tested by

no test coverage detected