From 6a4daef6ad3b1cce868eb7301679dd4a1c5f9cfe Mon Sep 17 00:00:00 2001 From: Naren Dasan Date: Wed, 10 Nov 2021 16:57:44 -0800 Subject: [PATCH] fix: Missing log severity Signed-off-by: Naren Dasan Signed-off-by: Naren Dasan --- py/torch_tensorrt/_compile.py | 1 + 1 file changed, 1 insertion(+) diff --git a/py/torch_tensorrt/_compile.py b/py/torch_tensorrt/_compile.py index e1a93daa0f..a6a0cb9b72 100644 --- a/py/torch_tensorrt/_compile.py +++ b/py/torch_tensorrt/_compile.py @@ -143,6 +143,7 @@ def convert_method_to_trt_engine(module: Any, ts_mod = module if isinstance(module, torch.nn.Module): logging.log( + logging.Level.Info, "Module was provided as a torch.nn.Module, trying to script the module with torch.jit.script. In the event of a failure please preconvert your module to TorchScript" ) ts_mod = torch.jit.script(module)