Unverified 提交 6134ec5d authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Model summary `pathlib` fix (#7104)

Stems not working correctly for YOLOv5l with current .rstrip() implementation. After fix: ``` YOLOv5l summary: 468 layers, 46563709 parameters, 46563709 gradients, 109.3 GFLOPs ```
上级 ee0b3b2a
...@@ -11,6 +11,7 @@ import time ...@@ -11,6 +11,7 @@ import time
import warnings import warnings
from contextlib import contextmanager from contextlib import contextmanager
from copy import deepcopy from copy import deepcopy
from pathlib import Path
import torch import torch
import torch.distributed as dist import torch.distributed as dist
...@@ -229,7 +230,7 @@ def model_info(model, verbose=False, img_size=640): ...@@ -229,7 +230,7 @@ def model_info(model, verbose=False, img_size=640):
except (ImportError, Exception): except (ImportError, Exception):
fs = '' fs = ''
name = model.yaml_file.rstrip('.yaml').replace('yolov5', 'YOLOv5') if hasattr(model, 'yaml_file') else 'Model' name = Path(model.yaml_file).stem.replace('yolov5', 'YOLOv5') if hasattr(model, 'yaml_file') else 'Model'
LOGGER.info(f"{name} summary: {len(list(model.modules()))} layers, {n_p} parameters, {n_g} gradients{fs}") LOGGER.info(f"{name} summary: {len(list(model.modules()))} layers, {n_p} parameters, {n_g} gradients{fs}")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论