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)