from local

2025-12-11 06:55:27 +00:00 · 2023-04-11 16:24:20 +09:00 · 2023-04-11 16:24:20 +09:00 · 042248251e
commit 042248251e
parent fe7bc9674f
2 changed files with 2 additions and 0 deletions
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py
@ -77,6 +77,8 @@ class Transducer(nn.Module):
            nn.LogSoftmax(dim=-1),
        )

+        self.prompt = torch.randn((100, 512), requires_grad=True)
+
    def forward(
        self,
        x: torch.Tensor,