fix: Missing log severity

Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Nov 11, 2021 · 6a4daef · 6a4daef
1 parent 8143489
commit 6a4daef
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)