diff --git a/ignite/utils.py b/ignite/utils.py index c45c49a24497..ff9f44afc4c8 100644 --- a/ignite/utils.py +++ b/ignite/utils.py @@ -140,7 +140,7 @@ def format_item(item: Any, prefix: str = "") -> Optional[str]: elif item.dim() == 1: return f"{prefix}[" + ", ".join(f"{x.item():.4f}" for x in item) + "]" else: - return f"{prefix}Shape: {item.shape}" + return f"{prefix} {list(item.shape)}" elif isinstance(item, dict): for key, value in item.items(): formatted_value = format_item(value, f"{key}: ")