Support --context-size=1 (#565)

When --context-size is 1, there is no conv module at all.
k2-fsa · Apr 8, 2024 · 98f9429 · 98f9429
1 parent f01690b
commit 98f9429
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/sherpa/csrc/online-lstm-transducer-model.cc b/sherpa/csrc/online-lstm-transducer-model.cc
@@ -27,7 +27,9 @@ OnlineLstmTransducerModel::OnlineLstmTransducerModel(
   joiner_.eval();
 
   context_size_ =
-      decoder_.attr("conv").toModule().attr("weight").toTensor().size(2);
+      decoder_.hasattr("conv")
+          ? decoder_.attr("conv").toModule().attr("weight").toTensor().size(2)
+          : 1;
 
   // Use 5 here since the subsampling is ((len - 3) // 2 - 1) // 2.
   int32_t pad_length = 5;

diff --git a/sherpa/csrc/online-zipformer-transducer-model.cc b/sherpa/csrc/online-zipformer-transducer-model.cc
@@ -27,7 +27,9 @@ OnlineZipformerTransducerModel::OnlineZipformerTransducerModel(
   joiner_.eval();
 
   context_size_ =
-      decoder_.attr("conv").toModule().attr("weight").toTensor().size(2);
+      decoder_.hasattr("conv")
+          ? decoder_.attr("conv").toModule().attr("weight").toTensor().size(2)
+          : 1;
 
   // Use 7 here since the subsampling is ((len - 7) // 2 + 1) // 2.
   int32_t pad_length = 7;

diff --git a/sherpa/csrc/online-zipformer2-transducer-model.cc b/sherpa/csrc/online-zipformer2-transducer-model.cc
@@ -24,7 +24,9 @@ OnlineZipformer2TransducerModel::OnlineZipformer2TransducerModel(
   joiner_ = model_.attr("joiner").toModule();
 
   context_size_ =
-      decoder_.attr("conv").toModule().attr("weight").toTensor().size(2);
+      decoder_.hasattr("conv")
+          ? decoder_.attr("conv").toModule().attr("weight").toTensor().size(2)
+          : 1;
 
   int32_t pad_length = encoder_.attr("pad_length").toInt();