Fixed logging level in distributed mode (#4284)

Co-authored-by: fkwong <huangfuqiang@transai.cn>
ultralytics · Aug 3, 2021 · 771ac6c · 771ac6c
1 parent 2d99063
commit 771ac6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/torch_utils.py b/utils/torch_utils.py
@@ -23,7 +23,6 @@
 except ImportError:
     thop = None
 
-logging.basicConfig(format="%(message)s", level=logging.INFO)
 LOGGER = logging.getLogger(__name__)
 
 
@@ -108,6 +107,7 @@ def profile(input, ops, n=10, device=None):
     #     profile(input, [m1, m2], n=100)  # profile over 100 iterations
 
     results = []
+    logging.basicConfig(format="%(message)s", level=logging.INFO)
     device = device or select_device()
     print(f"{'Params':>12s}{'GFLOPs':>12s}{'GPU_mem (GB)':>14s}{'forward (ms)':>14s}{'backward (ms)':>14s}"
           f"{'input':>24s}{'output':>24s}")