Merge branch 'master' into Maj-Pierre-1

export.py

@@ -48,8 +48,6 @@ import argparse
 import contextlib
 import json
 import os
-
-os.environ["TORCH_CPP_LOG_LEVEL"] = "ERROR"
 import platform
 import re
 import subprocess

utils/general.py

@@ -68,6 +68,8 @@ cv2.setNumThreads(0)  # prevent OpenCV from multithreading (incompatible with Py
 os.environ["NUMEXPR_MAX_THREADS"] = str(NUM_THREADS)  # NumExpr max threads
 os.environ["OMP_NUM_THREADS"] = "1" if platform.system() == "darwin" else str(NUM_THREADS)  # OpenMP (PyTorch and SciPy)
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"  # suppress verbose TF compiler warnings in Colab
+os.environ["TORCH_CPP_LOG_LEVEL"] = "ERROR"  # suppress "NNPACK.cpp could not initialize NNPACK" warnings
+os.environ["KINETO_LOG_LEVEL"] = "5"  # suppress verbose PyTorch profiler output when computing FLOPs
 
 
 def is_ascii(s=""):