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

---
 egs/librispeech/ASR/.run_v3.sh.swp            | Bin 12288 -> 12288 bytes
 .../.train.py.swp                             | Bin 102400 -> 106496 bytes
 .../train.py                                  |  15 ++++++++-------
 3 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/egs/librispeech/ASR/.run_v3.sh.swp b/egs/librispeech/ASR/.run_v3.sh.swp
index 9b0e1b66d7819b056f3f5b538eca201a96e559f7..cccf26bf7dcfb309160e033dd1860fca505ba057 100644
GIT binary patch
delta 30
kcmZojXh;xEG6?hZRj|}EU;qLE1_n;iDak1tMPKU!0CL?2bN~PV

delta 30
kcmZojXh;xEG6?hZRj|}EU;qLE1_oBqDaj5SMPKU!0CEKfOaK4?

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
index b5563203e5360e8809e20e6f721ad6b31baf1949..ba895242c6d41f8eea909a192aa5822ef5fd618c 100644
GIT binary patch
delta 1128
zcmXZaUr19?9KiA4)j4OI=9F-u;i!TAF-2)wE6wnsGLtgPw9;mqVP($$PSV+1NQHO>
z$t=pCEDhA9KGcI{^bmn*5kzedJw%w16l4)a-<cOa=bU@t-UIi1+iqyvCbUCAk&4M1
z9Jx_ZbZUw*CTgYpzbcpJ4MAR;@pi<BSu+%VC_<zY5-FlnjW5@D0(XNbL^>4og^Spc
zkNwC*2DU-LtU=@{deMx1$U!>NV8w5}$Vc48c^rWi$ykLr#OlQ(zGWh#xP+4^M;@$*
zLkOOPiS(cjdDw+?q```4{M3njz+<$d2sW4ygQ-xF3&?~S@tD?n_#ax4(>R6W$j54|
zz^@RIPq>FGaG?}wSch2+n_?1f9EK60_#Q0s4WoE~4z!{IxkyIT76t?K_#4E>@C*a!
zKs~B(6q}I_6Xq4~*Xe$P#mrSvZg$yg?3J}i&a)?~9L}23I=9Pd&d4w)I8qxDY~C?{
z&alVs5GhkcQrRhR{FO_2yjd=C3m)vj90A-x7gV&tgH~iC5nA{O;SoA<5pFc05rJ6N
z;xi#k;+lbjZe&r7Qk38r;;}#pQwWsOj}A1V1UsR@dm<P^2ioCA6B=<4Hdqjh7sPNC
z=TM9y7{fT|=Ty*!GT0G~*HmyHgXlvS94N$Atj8Q7yu%1?q6vXu5@A6I9uh+j4q!K|
z*n|Z__<@;Rh6xPdGOAI4JnTglEYRT#M>K<R4C6YQ;YKl1$E#D6BBk3^plG@$$`yk;
zpQQ8tnJ(9>+gn0ZhcAi;8nwZ<V{sn$nbcRl?7-SkRbS_-uPd#sXm)0(Eq;@l<qy2A
YQFHy~#rce1zw}(APQ49Pd;PKh0EcA20{{R3

delta 514
zcmWmBPbh-{9LDkAyUmN)^u8&Rtls&P6Qy?X?v>cdLDFIuM6nd%Fb58{Qf|_$H8~h3
zH#YA<c5?D37fOy=E;~rOkX^{v;q&ynJx~3T*J|=u%{w&3XFIhXMTw{Tbf3MbA2g;)
z#)aX^d+SByOCdCocv^q#NTeD?reHzEsaGV1C?*lcFovL_;t{#W2@;sV7%YTfLh#-o
za)uqu!bA}52*ALbTM|-qi|k_=ONhWk0B+o}Ifo5IFbWGHm}tRIok$rs*uxA4;YS;u
zYejYuMi)BqqKahIgvc7lDi#n$H(K#qBl3ZB9AO>v2%#4h7p1tv7N*e*52{X)ZxnHb
z47Ra?aRlLm-cM+7qU;cPKpq)vB91u>z=9wD5ByZJg_zQ0SDy|&*-}n1?911Y!XJBm
BO!oi)

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
index cc97dd06f..a700a5c66 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -550,13 +550,14 @@ def get_encoder_model(params: AttributeDict) -> nn.Module:
         num_encoder_layers=to_int_tuple(params.num_encoder_layers),
     )
     '''
-    encoder = FairSeqData2VecEncoder(
-                input_size=768, #params.encoder_dim,
-                w2v_url='None',
-                output_size=768, #params.encoder_dim,
-                freeze_finetune_updates=0, #params.freeze_finetune_updates*params.accum_grads,
-                additional_block=True, #params.additional_block,
-            ) 
+    if params.encoder_type == 'd2v':
+        encoder = FairSeqData2VecEncoder(
+                    input_size=params.encoder_dim,
+                    w2v_url='None',
+                    output_size=params.encoder_dim,
+                    freeze_finetune_updates=params.freeze_finetune_updates*params.accum_grads,
+                    additional_block=True, #params.additional_block,
+                ) 
 
     return encoder