From 427adade4cf5b4568ddc3c14c83c4ac593000002 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 11 Apr 2023 15:37:27 +0900 Subject: [PATCH] from local --- .../.prompt_tuning.py.swp | Bin 81920 -> 86016 bytes .../prompt_tuning.py | 1 + 2 files changed, 1 insertion(+) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp index d47d6c38cbc2de8a842b9467ff90dd198ed5e274..7a501e4ee9b239333a25edd61f98d82b43796e11 100644 GIT binary patch delta 518 zcmXxgODKe46vpxA^UVz7HXnm2w};&2VOCfvEAf?; zg(*t8mz69mT(V#zWhH;))UWrvo!)xR8;Lk0o6g+9n!ff%Si16~L0tV5p>{<#KFif6`8xWffv zSVkkN@!*njJmS@zcn42nwXw#>YADysj&M`w&%kG!vtlanEP+n|G&8NNH>UM z4J!zv6%BaJlkMfh6DUIwj?$$qETJ1!sDvY39WrTN(xgjlpdDYS(iu*% dgAiuWf>M;=lfdsd!YX2J+i;wv*l&q3*Efm$Rp%7xNI|FfXI4Fo9!ojFU0|yDgNd%k1e?TY#L%Jn*agDI4*1Dm= z#YM;XfG?`N{;sH$4{wn-E4F%_uVwQ zA8tB`C6|fL%zO4Cy?C5qX#UrYAUQ~z>s(CQ#Ud(w;}a9yp@RYr5jxTseVk(l2@Gwi zg$rcx8g(g>ALUyyLI3~& diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py index da810a1b1..9151e9496 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py @@ -801,6 +801,7 @@ def compute_loss( batch: dict, is_training: bool, decode: bool = False, + prompt=None, ) -> Tuple[Tensor, MetricsTracker]: """ Compute transducer loss given the model and its inputs.