zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed logging level in distributed mode (#4284) 

Co-authored-by: fkwong <huangfuqiang@transai.cn>
This commit is contained in:
imyhxy 2021-08-03 19:11:42 +08:00 committed by GitHub
parent 2d99063201
commit 771ac6c53d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/torch_utils.py
View File

@ -23,7 +23,6 @@ try:
except ImportError: except ImportError:
thop = None thop = None
logging.basicConfig(format="%(message)s", level=logging.INFO)
LOGGER = logging.getLogger(__name__) 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 # profile(input, [m1, m2], n=100) # profile over 100 iterations
results = [] results = []
logging.basicConfig(format="%(message)s", level=logging.INFO)
device = device or select_device() device = device or select_device()
print(f"{'Params':>12s}{'GFLOPs':>12s}{'GPU_mem (GB)':>14s}{'forward (ms)':>14s}{'backward (ms)':>14s}" print(f"{'Params':>12s}{'GFLOPs':>12s}{'GPU_mem (GB)':>14s}{'forward (ms)':>14s}{'backward (ms)':>14s}"
f"{'input':>24s}{'output':>24s}") f"{'input':>24s}{'output':>24s}")