[PyTorch][CPP] Fixes

a-sidorova · Oct 30, 2023 · 9606149 · 9606149
1 parent 7bab171
commit 9606149
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/src/inference/README.md b/src/inference/README.md
@@ -828,7 +828,6 @@ python3 inference_onnx_runtime.py \
     -i <path_to_image>/<image_name>
 ```
 
-
 ## Квантизация глубоких моделей с использованием MXNet
 
 #### Аргументы командной строки
@@ -889,7 +888,6 @@ python3 quantization_mxnet.py \
         -is <input_shape> \ 
 ```
 
-
 ## Валидация вывода глубоких моделей с использованием C++ PyTorch (TorchVision)
 
 #### Аргументы командной строки

diff --git a/src/inference/inference_pytorch.py b/src/inference/inference_pytorch.py
@@ -484,7 +484,7 @@ def prepare_output(result, model, output_names, task):
 def main():
     log.basicConfig(
         format='[ %(levelname)s ] %(message)s',
-        level=log.DEBUG,
+        level=log.INFO,
         stream=sys.stdout,
     )
     args = cli_argument_parser()