From 3e4c93cf915a7329e4c51a3846461dbe948a132a Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 14 Feb 2023 01:23:01 +0900 Subject: [PATCH] from local --- .../ASR/conformer_ctc2/.decode.py.swp | Bin 4096 -> 49152 bytes egs/tedlium2/ASR/conformer_ctc2/decode.py | 23 ++++++++++++------ 2 files changed, 16 insertions(+), 7 deletions(-) diff --git a/egs/tedlium2/ASR/conformer_ctc2/.decode.py.swp b/egs/tedlium2/ASR/conformer_ctc2/.decode.py.swp index 0f873fe80c844367024e08e98803778fb79cc825..b04292c6db7defb32b5cc353d6994b6e30e2e13b 100644 GIT binary patch literal 49152 zcmeI537n*5edo*dR!~{jE4uPh{Q;__ySjU35NK_5m>Fh9Ugxg#xs#sR}B!2#+9h(dJ#2dKWo&8C#V}&h$ig!_Z;t0RXqo2 z;=28Q<~P;v`#$IYIseD=f9EGJ-hM%FR&g@N^{8C#tjoSv`S!LA5BSK5xm>li6i%++ zuuiWIcWowI*sG0SP_CTQYVEtQ(<;_mm2$m5&I^5l6U(j6zNlTUgcG$&xKOUwC&I;O zq8nE0wO(W0#KzsbCMvDwLaWmVJEcmuvTmXpR$A4t*k0*RbnSkP0&5gFG6j0wh4C|w z&#hlKWrq4l{-og1|Nf#Q$6E`rMu9a7tWjW%0&5glqre&k)+n$>fi(*J#!#T!epv2B z)aKv08gFSfjui1%53lP_E{3n<&|Y_+|9}p8x;uBXhYM zKo^_|o(k@KL@xJ3@Gh_f&IV5h4+jqge|3B=_ad+X+;Lnk_Ze_2_%yf)yc_&3Xn|*f zXMoee6Torc|G=bw3cM4%11x|oU<}-Y0N}geOW+IOzkxpn{|UShEP_qoWN;k#0Rn~Z zf&UJ!2R(2pI2W7&o(zryKSeNcJNO*eai& zTsqWvP*^t_OwET;w-{_W(Cw5f-JrV^26_^-%iX15q0?%ZM{z>+a<_{~NKlUSJd<7! zG_U}v2Uvv!K729QPEws#wO0vEj36pELRXw9Sgw%?<|E^A_)LnFaB9@NU+jcob!9vX z%bm)SNwn3huLQj)3^wiA6tsHXcCQlW+1pCYNo6nO1 zVZA+*&*wMGI7j`U5q6ha)ubdkw`$F9M2DnLg{C7WkAtWkR%#11sy@FG5XVLOxaDOIy95LYK2O zSqV&H>M!co4OKa*Ze+%l9bvW(v}e+g%e8vlMAHCHVg=!Ww&|wHBSZFLJ2e^jUb1~5 zXt$!8$UjC|Ok)MLD5xzmG$6BvSFIZp+EZvUA;(YxshN*~Lie30boiu5J6hdvI@qF) z$LQ>-JDFK+DxqCm|8Cr;>|B))%{NMBZX2M)8D}K<4kd*l=`qmaXi2=-Y4zIU&0gbh zX$M@U$jO+L0iQ-bHhlGR6wNSr%}_3Gbfb8)ogboYOeE*}z$OtpNb+u&YVpBwA-VG1av@PHn!|4L8>+3}Mp?oONT7Ij2rX zZ)h$J3fXs^X0udp*h$_8+0|^G-?%`_T{f&=OI%9l+odHNCT#Ba$*6?fmTH(%I@+*O zS5$E=jiu$KRy{1aHP@0Q^WGM1$ctu4i+HJCi`-457SSwp;VQ)pc&fQJbaZ4uAZjVk^yEDUZ^#zrR1UO7p`s7FD!Y6HTsaieQlkj zpOh(6V3za}dw}**snTk+YjxHU*PiFJ!%f}Fo=&;eWcJ;vFq>QucEj2%#a6LC$|i3n zYu}fnPN`b%mg%Owdgue~39B2U&Ssv|r+A)o`E^^`?lha)ZZ(^kr~bz8cFbl+;i7iK zzuSjpuqhRjkyd|~s_Fm1`RW_N&bAi)a(yh=(`%El39h|gy*BT|U&yVmU0us0G2+4E zrl=}o%4e&=Np}9$PY?|Mk)U2{)EMz5gWc!|NJ1^Q3T(WW3u8lkPY~Qjw*tsj%v~$Pi z?R&QG+`-+s!Nwhz2Ipzx;GB)Sx9=V!jfMTx>mM9oaoWFh3mL0pdjIYCD zTv4=~Hp(kOYradbWg}T>wYghJ(>*A63Ay4zHd|)XNdPLhK=)@8Q3=~!6-xu6UMp*N zZT9YcjprIIO{k5gjJ}1(f{VkB8Y-ZCK|#^;E>eER-UNX=wZ)}wux@hQRPdZyxz(ugw8P?J3tnLFNinCP%8GK`+G`Q;V-mjrzyHVJP2do?6g&q6;5+d8w}TtOE5R$k#b5$F z4*UZC|C``u@HTKQ=z(W}v%r(V{lVSjc^9}Iyczrscqw=Z*aw~kwu2%t<#T0Q`&pyF z8U@xUuttG33an9JjRI>FSfjxIXA}@?(1778jOJtGU`-5gg!YFEK$S~vY#=IT~33a7Ax`n#pF>osH8zgs5leGmw|&!sr$@w9-IB`Ul9Mug)qTJY*AH;qo9ioeKdp-|Nm$v&pTxZ z5C8A&=l=md|6|}aU=~~kir{Ye`@6tb!B@Zs!Eb}Bz*B(u|Hp#Ifct~*z~_Gsyd3;4 z*a;pBehyFnWAJ5gGk6u)4D#Sx$Rj=jUJ70UP6tl{pGU9%R&XWQ08Rn-13B4S4)%e|zz(nhJQREn9sS3_b>IM427AFX zz}et#^z+i;-vn+1*MK@W9c%!P20unO|JUFaAl?0&K@(gA&ITuf9QddELl5wN@ILS= z5P|1_OTdG{KcdI~8u%c1J-8C=2V20yfDUq~Ex+v#Ht>i`JuQ80qaW>Hip}X^SdUUl z-FO|!EnnZ+gjN{gt` znxS-`2~u}9?Ksh_CZU+hBTl?8NGh*{{)%Ubn14C}Cb&P_Y5P-|Q0`$WoU=yA1ar@lP`sr~vL`V=vMBLualf7{V{u}3okX)K*(6BIRf`u?OFC)0 zRi4g0(`naQQub9 zk}bQW*kK}G2s_N|>ON8;S)-s-`TCMF)m9aCwL5SI1y`#$b6@h&F{2ThS+E6Agrlff zDqUZs=XNPA177+ipPJDqwc25`fN22ReE1BuiOcy>sWlfYjp8@9m(W>@zEe#E3wek+ z9`zcHa%W}yV5XXfiroWU+jjADk)Cu|@)&aQme;gM5eo$kgfTVROugKgua<)Y)4_q+ zsl9G=s8+f%9VUNCNByK4qHU|A=`%(SfiZGmN=vt=sfns@+B2aib03eG>SW~L7Al*O z+k5*?GfT<#QhaKW^8xgmB;}tjK<9iPImQnX{SYnSI0dwzajrATpE;j~e#uGs^;>Tw z3wB=%`q~Y&q7uv0e|T#uNPjchawm^MW$cm1*vUyfR1`smT70<~H_H zjPipWEbkUEf$1)36*LL(Utb&h$0lCdFHgtbS^Pne;S@w5QV>w&Bf$G5gDykg^v*F^%17 zD==?@s3zsKm{7FApxbIO<2P5-Zmwqx>?5f_VVDJEkcUwpuQ&2+I^4b^UD4@4mK3wH zj6{3q41`P)JBgT+~yd0k=q3?=i8bN}#~Hl7z6VU>T|w3d}1d%@0f! zOfUA&6hCF-scbG7WYOQMLovcqBHh&$SY`47>@I-H+JWNJit{0@*}X>pkE_4jO^*Mc z<8NJ`h1Y){cp<2OCxP$6&)*6Tg3G~0Ks^38;OD;tZUC~mj@bfo;4wwdG z;E~`5@b2#eSAu7QB6t#b1UMet1@C@4cqeFpI@kl&fk%RG!MFb@cmX&MJRUp(d>=mj zv*5)b1ebvI;AHSq`18+#8^FuK61WhY15N=agFE5RzX<*e{3$p9Hi2`%J+#5iKyC3} zwZ|%#nvyY|)325Ec&Qj|ieXi!hBZ@&WtW)|9)Q31Z}Zq#(w& zcQ8=8M1ya$xwxHdHds0Vwf&&AYqI~!3W`*XbnIfcRj}2~k|BH7A1hlLy-gTVm%1zW zp$S8Sm?QOXWqW-Hi}RGYLdlq#B*vIYEt2)wr2zG{YD^^)WwyxO8uqfGOs>4j!M9uL zO|~9db?8no08UBaV*653nyvRlfIe<`=b=r_PnRnS}`*F-VUy5p1 zggzg$K*~HjboQ_!Ubxo_*RBR)uBloEDa zgy@bUB4?<8$X~-@L`47c9BOR!_X{!ZLN7G9r5;Tew0pO6pl(c+ZU-(YagU~ ztqG=}y0U#>_!B*~U4E$4Qm=LSZ8pZ()tU57wryFiN$WRFZhB2JxvVm)*kWor z8J>)(<6#rBR2WD>BMj?*8~<#k8;oxyjmu=?Q%}<@8276M6ICX(OVS0_GqwI|{&Dakghm{7D{#@ z&j~7IL#^H-2dGkq@(KsVt;J;;J)|if19?sKSyCt&j!bZ-?P8r#+@foC!@1*ipe%XFElw>Di6*^!f}TN z#i706=KW*+`eD85QlegO_RR4eoz3?^S>Ef2<`{I248~rEkq3MPycfJ1ybHVga9FJdK3Y$#yjs-xlXE}i6eZ_{$!JcD zxx;Z=OBYl|_8JfcFSAYdMudmv21jMPgS9S$gRx3|@bku%O@*4vc17I_hyGaGX|VWgg=UVV?&H zREtL7z$|SN zV!xY+1z7j2qX9_J*z>bmcBl%EbuYQoEJ ztaN+E6N=e6h#IES&PR{k-X?nyo3=Itq|fM17{0ub1}VOcladIfBV|QUuBgRSGCdTz zi{1+f9)A{$b{O|Na;~HG$r2_xhU|C|FO*7>vaNN=60UD(3>sg==#ya;gH64TY-Vr; zLk@-pk#(&Tr7Gu#yHpi@ZO5XOa$cbcjy0Ld2pSpbUv=Ej^H>@%z3H$sW;}fjjHTi4 zjdvN&h}XjXc)Ve@UFdoLO!GZ+NZ(D5IUC>pk87ecM>7+)JT66n?29yfXj|+lpyj;b z_nBig;o32uNhmNg@)8t~i9`)k3PSpAl5ODPgxlem0Q!7bUrEjw^}E(!mtSmbqYo@N zrYG7>STVSS89YJ+-R}U(`=~J0Fgv&SAo~&;1Qs<`0n9giO-(mbSA_P%TxdNN>!kjb zFL9^Uw5z9;Oo?#p4{al2R+6qMnsQ6o!{&z-n)gnHSgW{LWNi&L_(vEkDGRgQnVQiu zJ~MVg#&K6BQyTj8(ZDzvVSX|1lNG(_=$)DxMFtnnH9<3zo>b^zaO`B10^Jgw_zj=* z(bP3=z)U%)*9PB#ZOWAVFpUz^+;@&lYS=x=P1(Jx8bw*}igFAEcv{>$4gW9J|DPTI z@BR1R1KcD>;umS&j#y&`~f@+d;___-$^zAJ_gk(_jMpBQl1+0v`Y`2eLPq2hRkjf?ptO z_!{_I@D}h~pmPQP1Np+6fX)EI{G4*m{&4O|1Zg1eCu+yEp$*bYov zOI8r{U&hP@8>cIn&gaZ-c(%58Zk4&iEcfcRwQ(G{_@u_|Zhu;y`5ccoeCA=>i-6Z; zGCCLFx&+I9md`jLm*AYYu_;X!&zS3fiwJ_bbnd``192%Wv%%aH$DG3(TNA#yjB}At z2A5Q*KOr^%@!Y9-i0P8mwQ8q48^sJ7)?vMl(=I9NWUY3EhFNH~+^lmnsWn?V+NO0O zwfP@cQ4Jm8HRP0plK>*OvvD=$Gi~7Kmtt@(8!%VTG26TebEDm2xoc&P-e(^?5PXnr zb?vj9$(M4WJgi5ON0#9zEBJ8S!&ChgEIIFtqrv#u{WfaGPz@qsjApaiIvqO!n&lM6 zxlvh0rP2pms@Vez(`4k>8GAe-^&mMpH?4CVyH^_Xt@<9Bu-X(%@uyAE+yj54E|vA_ z$5EWo+_qF;S5BK6r$knJW%m~L&|NA_vst{r_*rsEo1NJ-9+uo=o35414z!%AWMg-S-SoI<(8<*)`bM|y&~ zh!baz*v*aH5=HnWT4b&4%@=VLG_f2u!|r&vF0_A5%-37<6SBT{_r~cO-;r%JT21Zz z*j>}B8!!;1D$c0vh$=ynk~y9s3Z+V5I(J+i%=t`7T?-HtlRe9>o9B9O)a_iDVmog z8SM+JB{UfIQUMR9xR>j2Ryf+Hb1|}VE}S!_gDmtGoAGK1?XQK)1vBc5AxRu*)ihKi zW>Qto6oqo>G?pGmRH?nMfHGXBFF1_Nlw%BqhoM0noTyEjz2sc7_JvNYTSjer+3FC< z*Oz!jt#5coH3|FYq65Ji^pDO;iHR4(6Kh_hH<_R^@Ae_hixCd#NK#}Og5Vt9Ce$$@ zlG`|DFsJ zMK_lWpj1^%$yXurG(ANA=xmAYBx!>Vdz0z~{jW99)nROTI2f+-^v$7m_BF+EwSw@6 z%sqdYEAL$==BLdmAAifKjhqW&!?95)Ra-rssjwrN2w4_h;qoFUK9F&LYP?d05y6m6 zrKM9g9H;mD!>vz=uuLG^|M90-#^NeCv#&(9%Kq)TzQ`k118l6LG8>4TlJ-{~aSR4a zcA9YFAqw1|l@CGppdhDDrF1zFfNgo0IC-;Dgx#Twz=AtIXgtAgzj>P z`c2PjO?jt|js9#}X07LKoTe8p^_`&gs)#}9lGFHL<2deeJ)CqwGdE0D-;B{s2r z$douOt5QT?wp804PEja%?J)~1ZIQMaJ4-D#Ui`mWS+d>qaK*7Eny5;=s>ybN!|VKP zakh9X&R=ec;=XXiLh1bfNo;O@==lFSf9v`@JpC=;&%sN;i-Gw4zlVSSBd`su2kXGe zK;HoP8NB)zz%5_}w80p7BKRBl^iP7M1n9ykts8Gigz zKxh6>2jk!|;BNTvJHc(>gWz>w1?&O4fougv!B^qSZv=l1WGnD8uos*Qo&e;({}%8e za6RaOXMqv$AAo!S{0sc~cYw|d$bR4h;I-fnKo{%;8^C(-c^Y2QC0l0RI4g z{vPlSAUlC8!KL5?@L=!}c=fBn%fKvH2V@WMQ+W1Iflq=LfZgD!;ECY3fZFZ{YQH00 z8fxs*?M&By%H6J(&Ii6f!mb!@&#MD{(`GZ8ER$l4IU|}mB%f0pUE1IBb~f@yRm@t0 zXV>)BR{=+CP+0CV#}_m;Vy)$VrK}CCC~|i8RG95k$AVK=R#xZn7&=3r4l(ou6BFPRHR zn897v5-X^V-SO7KxUb^&&51N(B&RCq8gWO>IIEz!--@%hnIpf(ae)LMiOOdLj#`e$ z=i>Z}{rxpLwq-ZFC@p_9gZ^mXVG_lY&U%{0I^BMu!QLjpQoHDx z)ZaFMYj)Sa-Ce=%Y}B;bLruj5x)>%hTG&fl6- zY+WuJmqI23I@LCxtYcP2bwb7R7|0mbuS>eqR6>!y&3)gkGQ}A_@;J7f+BwYRyPaOM zl2Ualh%PtF9i*{ikvcHH>K{|03=#)TzQ%A+h~ zwvG|1qG?(hCjV^S7o$Y5drM z>4#M;IyR|TWt`m`pm?<+V!H@!$G{acBuzOIS+_Luml!Zcn66`W=3KGHyI;VpEW@w{ ze!M#YbkTThFV7>jBSrhE2A$25seyPOE7fMlUg9%)>Un@K07 z-t=V&WPg@lMQfQyRo!H{Z=!UUOEOA8I!%$vQ6-SBfv@{W|3U7ac;w8&o85WwI^SpK zU2-&F&sT1?uh7oOCwmPGfNHB^ zKJH`AKEyd%LW3yuHLA-=z zQza;5Zny4TqT8~G#eLrRMmlnmJ9h3F`dKS_VK(7r*rT7?>YE|s%oiNBGWrYm%yzhR zL{AAbJ-VgTQae$8$C2Z+N{LnQs$ zhp18oGvV=3!fa)VVdKZGA-z2Nwqr=peC;fipt+CtNk-8>m|$F4V{SOm&tZn97uaG< z&m|s|;(=znux)fJmbERD`oE_S@R(03Fi$D$p9V+ZYLvs2k<`w z`2*Munc0MnN^-=v$TO+*uGc#pwo<*Mzb#@X*c$#afSSYY&hS5{q2>6in(ob zp0~dpX)U&-;v((ljAL>+kxAD`Fqdibfv|Hq6V3$Pvvc!Kog?F;Gjpy`*k3fGE@FJ` zQG8NykD6n#jN@n9Gq&%LJY*&@4Dh-A2KDTirtw!0SL@_iqGaxp#ukO{6Lk*XTQXk< zv!i~M5xd%Ed9~W5#%tW;O<9cfdbWa@y3b_pvH_2F5DhqM_M^)ECaE5tJgIM>S>g;8 zW0|TZR*znim8pjeH)f}b8_;%R=g)sn)qiK`zccmUS!4Q~k!9-!BTg#NqykM9Is83U zM3+8QT&Mrm4~s)ylL|AbFq6gg`j4!2#DM?E8ape;#3r0%-YSDqVvhV^XHF=$`j#)A z1O_-C8{2Wb*;Rp!%WN`JIWAzBJlcz#XA{l^12(hsHAgFv{<=F3Z;-0OCzW)uhYEA! za%<38vG+ngG20rq?QHsQ?4Tso%?~9H{nUOauSvJry1bl8J3Bclw^TAm94!hzAdB^} z23{l$rg*xJb`UA-s^dm4iJ>4OzRiz{a_i6q4GJO zhJMqCaQ^T<#+xQxftZO>?UuZq>}2@+&6v)`n)#rhA!i_hH%%xEZ`1ycjHjT|oW<9|eAf9N=T%zW~_=TnXeiumDZ~ zj{*+@zd$DN7eMlYtAJz$vh6V{u#CD;lc z0=|e$;1l3V@O1D~y9gpGC=(|RjpTEJ0m z*9w_VN*3|K6<-B$=^x(pX3iEl`+rgaZo)>L|NP{S`D#=WKjOaATi zU}SmsoqhpQ{q=zp8-#IZN1ydNn(#wPNmq)>W;T&_MR28Y+Np@PgchZyrek)nSS6$AH7=On6pPk@hY5cF{*K!ha z#-GGV+wl8_bbUij$e+cUJvhssYA{pjex2ehA247aldvkLe~`c`q3yOOrwr9COem>Z z$LycVCuAQf_dQ|3e0xVh-8dv!XOAbPUO{4+(PYD_?Wai(u|4bjGwk4YpYYwY(zgDW zlAfLouf9!wIc0}#)O9t5p@K@#9tcRxp*fDAQkdOJ`hesfHb8ccq32qex$#P0VUMM= z(dL`()-{0MMyQ?GHrcQy_rRPvwW=f@%PPlO9`Q*8zNXA+Iu5o5<|bu6v&=MC4~+qa zuOg9Qte9~iG@?#@44LEoc|R0&YzdD8!;cqv^~5f#61697q@~mbPWykKTC?S32%$-f z!$s-SN~S_Kjo)D1r+V_yDn1qVFj`NRFxPD~83jkH`&6e}Qz!m%)rp!3blE1c>;a>l zPnaK_)=P0OT-l?O102DSy4g4*V3D>CW_0{cGk{Y>YolVfEPm`&>vkk(Pc>{d)(bvX z(Vk}5%uJN>zkXzY>^9{g=f{}o5=r+j;RxLS%O~E{f;0jxqthSe)Rr+DD9S+Ke0i3@ ze?2RbYq#Vpb9Pa393^vpPATc)Z%5#qZU=Xr1c-N=a#*kPW?H;Wa7EFj2Rxs>cf!O} z3PS~oCJX*cZ`)K9mS}H6YPJLZX2>RvMI{b9WEZC#EG5i3o?64*uG5|bzy;$)O-z{h ze7u&~Sm0pwKzda&Q4S z4g4Iw{6_G0a0Spe{Ei2=!k6C&UIv~Io&bIh?cW8|2G^+_{N?9}>}b0vPp98cf*UV&7aGqyU#fbhby>liNQjz_& z@EhTv*7F*^R{dgj|NNIy+g}&wv^ifZ{l%p7dp|lULm#e>=4% zU2PF#@V35Z>{q(@L9zm9_P1NkPTw~H2H#|3`LmVwteek5v9fQHScBH)KGxuyB$iTi z-t{i_5!QC^LxJ1k>L-THG@ntr7HY+3x zS>{8X+39erz84supx^48=oKrNLGd}qxD#CGeYnsZ7!dIA&g4LMP;dAczxW z7oFr#?aBW~QZV&R6>e+4>9$_z(o6hFa0I!e55X0-CPhlWu2y5)tthN7SOc#9z^Ij+ zEtNQ$tkERuozwwRB|Q>stn!68YY~P7q|T8enT4rxTcuy(XpwPKhl_+Z0*9O;=}LKn z(Cwh^-#k5>{@(;x%_=1677Vw{=29Epl%ij$IP&2(q$q=5yYf`CG^|2m%RFdV^lqwY zGksW+Z`H=mMD>jjL@I5li#rwUpxjp*vdLwbb+0+jGwg#iCQ`9v`bw!Zo0p>r zPH1qDnZps#FFtAH3!&ekB*bzZx+l>!bXJMI3Q6~^8>hSz`_@f3VxD=%hFm&d{mlep z4(iSNKja?VNH>jofr@aqI2;9IWABB42~k7eR3IL-U$~S?^3rZe((>2}goW8!y_kedi9m9xyE> z9*1w_^dhotVwLB*Ueo&VLB7v-=;m07JG=>(LU7I5#gwJEJPDsv6o z9+hIqqmqddQ$(Lpe;?ClLUBD@SSX?-=i{wNk}H+4&G4{apdK}jyXAdhaAwe`HGBAvD&~{f zN-=#beG9b^Pj&I^7H63A*|-hT;pf;djOJ>_5 mwSpzz6p&A7En|l!l-`loFGA7dtpBRif*BVs)}Q%l@c#l&r-obr delta 16 XcmZo@U~W*@@RE7r0oKj$n1y%%In@SO diff --git a/egs/tedlium2/ASR/conformer_ctc2/decode.py b/egs/tedlium2/ASR/conformer_ctc2/decode.py index b99c48cfe..65f99d9c9 100755 --- a/egs/tedlium2/ASR/conformer_ctc2/decode.py +++ b/egs/tedlium2/ASR/conformer_ctc2/decode.py @@ -28,7 +28,7 @@ import k2 import sentencepiece as spm import torch import torch.nn as nn -from asr_datamodule import LibriSpeechAsrDataModule +from asr_datamodule import TedAsrDataModule from conformer import Conformer from icefall.bpe_graph_compiler import BpeCtcTrainingGraphCompiler @@ -731,7 +731,7 @@ def save_results( @torch.no_grad() def main(): parser = get_parser() - LibriSpeechAsrDataModule.add_arguments(parser) + TedAsrDataModule.add_arguments(parser) args = parser.parse_args() args.exp_dir = Path(args.exp_dir) args.lang_dir = Path(args.lang_dir) @@ -960,13 +960,22 @@ def main(): # we need cut ids to display recognition results. args.return_cuts = True - librispeech = LibriSpeechAsrDataModule(args) + tedlium = TedAsrDataModule(args) - test_clean_cuts = librispeech.test_clean_cuts() - test_other_cuts = librispeech.test_other_cuts() + #test_clean_cuts = librispeech.test_clean_cuts() + #test_other_cuts = librispeech.test_other_cuts() - test_clean_dl = librispeech.test_dataloaders(test_clean_cuts) - test_other_dl = librispeech.test_dataloaders(test_other_cuts) + #test_clean_dl = librispeech.test_dataloaders(test_clean_cuts) + #test_other_dl = librispeech.test_dataloaders(test_other_cuts) + + #test_sets = ["test-clean", "test-other"] + #test_dl = [test_clean_dl, test_other_dl] + + dev_cuts = tedlium.dev_cuts() + test_cuts = tedlium.test_cuts() + + dev_dl = tedlium.valid_dataloaders(test_clean_cuts) + test_dl = tedlium.test_dataloaders(test_other_cuts) test_sets = ["test-clean", "test-other"] test_dl = [test_clean_dl, test_other_dl]