From 552f89adee291aa272e5741566719805ab298c96 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Sat, 10 Dec 2022 12:49:07 +0900
Subject: [PATCH] from local

---
 icefall/.utils.py.swp | Bin 65536 -> 65536 bytes
 icefall/utils.py      |   8 ++++++++
 2 files changed, 8 insertions(+)

diff --git a/icefall/.utils.py.swp b/icefall/.utils.py.swp
index 5ac65ead6b142fd9eaad1184efdb5e407783efd5..4d362b1df33f9e876d391984877f2a683cee28a9 100644
GIT binary patch
delta 446
zcmZo@U}<Pz5lb=%^Ym4))H7fJ0s#hwhd(AK2i)2yCMCcav)NGKvOVuaCI$vs7Km)j
zX2F19M#Xkk1_lQpwg+N9AZ7yMS3o7ZfH)0^WjD75&SIKe)y-9(T9KSuP-3N!SDG82
zR+N~VTCA5?P>`CJqETF0kXlrhS)7@lr&pAkoL`iZnU}5yQdgXqTac4ltjVB&1WJl3
z37Jw-l$e}d9BmM*mt0x`G1C#Zp=tR=3P^5H$jn3L7DuZC4N;G^ntX1Tm`Zdk*ey`c
zfZ1s7QLvrd9U>jifUHiTxFoTt1mbN4TZL!?1(2PYAe)L3^U_l_a#Hh<gP}ND323KM
XtfnT!I<OTeW=ziam)X2C)O{5IJkyN=

delta 122
zcmZo@U}<Pz5lb=%^Ym4))H7fJ0s#hw$_JB^k9lnrlM-MI*=#6q*`C*zg@IuwD?~PA
zvtU3lqaujA1&G^$I2MTAfLIcUzXFw>1LFOgTLWh?O*Ze~2V&vP9ezoSlN<F#85uV3
I40T@x0MpDKQUCw|

diff --git a/icefall/utils.py b/icefall/utils.py
index caaba6aea..04482b8af 100644
--- a/icefall/utils.py
+++ b/icefall/utils.py
@@ -191,7 +191,15 @@ def encode_supervisions(
     returned tensor and list of strings are guaranteed to be consistent with
     each other.
     """
+    try: start_frame = supervisions["start_frame"]
+    except: start_frame = [0 for i in range(len(supervisions["cut"]))]
 
+    try: num_frames = supervisions["num_frames"]
+    except:
+        num_frames = []
+        for supervision in supervisions['cut']:
+            try: num_frames.append(supervision.tracks[0].cut.recording.num_samples)
+            except: num_frames.append(supervision.recording.num_samples)
     supervision_segments = torch.stack(
         (
             supervisions["sequence_idx"],