From edc4eb3372e64f8a3eee10262ad9fb278babf660 Mon Sep 17 00:00:00 2001
From: zr_jin <peter.jin.cn@gmail.com>
Date: Mon, 21 Oct 2024 12:33:28 +0800
Subject: [PATCH] updated utils.py

Co-authored-by: Fangjun Kuang <csukuangfj@gmail.com>
---
 icefall/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/icefall/utils.py b/icefall/utils.py
index 7b1301faf..0682252f9 100644
--- a/icefall/utils.py
+++ b/icefall/utils.py
@@ -2287,7 +2287,7 @@ def time_warp(
         return features
     assert (
         len(features.shape) == 3
-    ), "SpecAugment only supports batches of single-channel feature matrices."
+    ), f"SpecAugment only supports batches of single-channel feature matrices. {features.shape}"
     features = features.clone()
     if supervision_segments is None:
         # No supervisions - apply spec augment to full feature matrices.