Logger

class tensorrt.ILogger(self: tensorrt.tensorrt.ILogger) → None

Abstract base Logger class for the Builder, ICudaEngine and Runtime .

To implement a custom logger, ensure that you explicitly instantiate the base class in __init__() :

class MyLogger(trt.ILogger):
    def __init__(self):
        trt.ILogger.__init__(self)

    def log(self, severity, msg):
        ... # Your implementation here
Parameters

min_severity – The initial minimum severity of this Logger.

Variables

min_severity – Logger.Severity This minimum required severity of messages for the logger to log them.

The logger used to create an instance of IBuilder, IRuntime or IRefitter is used for all objects created through that interface. The logger should be valid until all objects created are released.

class Severity(self: tensorrt.tensorrt.ILogger.Severity, value: int) → None

Indicates the severity of a message. The values in this enum are also accessible in the ILogger directly. For example, tensorrt.ILogger.INFO corresponds to tensorrt.ILogger.Severity.INFO .

Members:

INTERNAL_ERROR :

Represents an internal error. Execution is unrecoverable.

ERROR :

Represents an application error.

WARNING :

Represents an application error that TensorRT has recovered from or fallen back to a default.

INFO :

Represents informational messages.

VERBOSE :

Verbose messages with debugging information.

property name
log(self: tensorrt.tensorrt.ILogger, severity: nvinfer1::ILogger::Severity, msg: str) → None

Logs a message to stderr . This function must be overriden by a derived class.

Parameters

  • severity – The severity of the message.

  • msg – The log message.

class tensorrt.Logger(self: tensorrt.tensorrt.Logger, min_severity: tensorrt.tensorrt.ILogger.Severity = <Severity.WARNING: 2>) → None

Logger for the Builder, ICudaEngine and Runtime .

Parameters

min_severity – The initial minimum severity of this Logger.

Variables

min_severity – Logger.Severity This minimum required severity of messages for the logger to log them.

log(self: tensorrt.tensorrt.Logger, severity: tensorrt.tensorrt.ILogger.Severity, msg: str) → None

Logs a message to stderr .

Parameters

  • severity – The severity of the message.

  • msg – The log message.