TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

❓ [Question] Why import torch_tensorrt set log level to info automatically?

Open KindRoach opened this issue 2 years ago • 2 comments

❓ Question

The default log level of python is warning. Why import torch_tensorrt set log level to info automatically? How could I set log level back to warning?

import logging
import torch_tensorrt

logging.info("INFO")
logging.warning("WARNING")
logging.error("ERROR")

stderr outputs:

INFO:root:INFO
WARNING:root:WARNING
ERROR:root:ERROR

what I want:

WARNING:root:WARNING
ERROR:root:ERROR

What you have already tried

Below statements doesn't work

torch_tensorrt.logging.set_reportable_log_level(torch_tensorrt.logging.Level.Warning)
# or
logging.basicConfig(level=logging.WARNING)

Environment

Build information about Torch-TensorRT can be found by turning on debug messages

Docker image from: nvcr.io/nvidia/pytorch:23.05-py3

KindRoach avatar Sep 23 '23 13:09 KindRoach

May I know how are you compiling the model. Generally for specifying the log level you need to specify the log level like this -

 with torch_tensorrt.logging.warnings():
        model_trt = torch_tensorrt.compile(model, **spec)

apbose avatar Oct 02 '23 20:10 apbose

This issue has not seen activity for 90 days, Remove stale label or comment or this will be closed in 10 days

github-actions[bot] avatar Jan 01 '24 00:01 github-actions[bot]