From 52ae2ba762ec8bde32a40c1d7cf08c6b622dcb1b Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Tue, 24 Jan 2023 14:09:46 +0900
Subject: [PATCH] from local

---
 .../.utils.py.swp                             | Bin 4096 -> 12288 bytes
 .../utils.py                                  |  17 +++++++++++++++++
 2 files changed, 17 insertions(+)
 create mode 100644 egs/LJSpeech/ASR/pruned_transducer_stateless_d2v_v2/utils.py

diff --git a/egs/LJSpeech/ASR/pruned_transducer_stateless_d2v_v2/.utils.py.swp b/egs/LJSpeech/ASR/pruned_transducer_stateless_d2v_v2/.utils.py.swp
index 261bb3d479819287a1214f17c36de162d7c40d5a..ec472703097204a388304095d9f343573b94e81b 100644
GIT binary patch
literal 12288
zcmeI2y>8S%5XUz((1g#1av+WGbm#S5(h%v2AQUMmL68VatUY_3TV?IFcGpKp5#E4T
zpyD-n2;ePfk!YFqos%dE9R)$N(%<X(@a*i*a$(ui<ju|=Jc$#A<8C-C#;==<<z^<6
z^^I}o`_udk^JJrjdm?*k%+X70Vr?>^FXx|j1^GK;k6bM>$yFu`p*5F>j_*D{sHMzi
zeCuGJ*S4u-o_Z@Pmp7TTsq?~1EuBmA@jRW6dE=FKasB@CI%E?8B5)OfRkl7JwU7GR
z<45q|{_Crf$VUW-01+SpM1Tko0U|&Ih`@hJz}GA61Mcm{a)0-h^U9@pY0^XlhyW2F
z0z`la5CI}U1c(3;AOb{y2wXz~B4_OFDq~+z;roC0{{Qn1W8YEVP+w7>P<_-x)C%gy
zZH%KnqduZeQ2VG?s9UJE-ZkVwEfWDEKm>>Y5g-CYfCvx)B0vQGO#-Lzmj(-Utn`ho
zU^}iw9vu%L>f=djio!`hfMkFPrBGEaZEpZ`p&L1Ak7%7mSX=4*I+#F|40_OqXgGo)
z<f;tD7}~rGtKxb|>r(}kv2v-ZygZaP>TR~}vlXynDf}$Xq|#AatZcMbwU$`kyOTxb
z64Qdgp^LUvf@=cDu~R293ca*mz_Hx?3scDpog9O55PXX!)?lY{wX!mY!kQ9h-q&uE
z^F!rl%{0zT$#s*d+#)8&+n?Zv!h2bHWhy?^X39(9ytIp0dLC<Y&u#pxeQkGR#By0c
yfay)z?IYZo?tBn<PKKlIb|eUkF;&*adho_(v$(3_qN%dbfY2ZuY<KZMk^KgHALl^;

delta 6
NcmZojXi!)n000P<0x<vp

diff --git a/egs/LJSpeech/ASR/pruned_transducer_stateless_d2v_v2/utils.py b/egs/LJSpeech/ASR/pruned_transducer_stateless_d2v_v2/utils.py
new file mode 100644
index 000000000..91ffcd85c
--- /dev/null
+++ b/egs/LJSpeech/ASR/pruned_transducer_stateless_d2v_v2/utils.py
@@ -0,0 +1,17 @@
+import math
+import torch.nn.functional as F
+
+
+def pad_to_multiple(x, multiple, dim=-1, value=0):
+    # Inspired from https://github.com/lucidrains/local-attention/blob/master/local_attention/local_attention.py#L41
+    if x is None:
+        return None, 0
+    tsz = x.size(dim)
+    m = tsz / multiple
+    remainder = math.ceil(m) * multiple - tsz 
+    if m.is_integer():
+        return x, 0
+    pad_offset = (0,) * (-1 - dim) * 2 
+
+    return F.pad(x, (*pad_offset, 0, remainder), value=value), remainder
+~