Fix a ref_channel-related issue in enh_inference

espnet · Feb 28, 2023 · 4e3e14e · 4e3e14e
1 parent 84ccae7
commit 4e3e14e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/espnet2/bin/enh_inference.py b/espnet2/bin/enh_inference.py
@@ -172,6 +172,8 @@ def __init__(
                 "Overwrite enh_model.separator.ref_channel with {}".format(ref_channel)
             )
             enh_model.separator.ref_channel = ref_channel
+            if hasattr(enh_model.separator, "beamformer"):
+                enh_model.separator.beamformer.ref_channel = ref_channel
             self.ref_channel = ref_channel
         else:
             self.ref_channel = enh_model.ref_channel