Fixed configs

Masao-Someki · Jun 20, 2026 · Jan 29, 2026 · Jan 29, 2026 · Jan 30, 2026 · Jan 27, 2026
commit d3d89c3a0f30eed9943793bae8f1b50a650fd7f8
diff --git a/egs3/TEMPLATE/asr/conf/infer.yaml b/egs3/TEMPLATE/asr/conf/infer.yaml
@@ -7,7 +7,7 @@ data_dir: ${recipe_dir}/data
 dataset_dir: /path/to/your/dataset
 exp_tag: asr_template_eval
 exp_dir: ${recipe_dir}/exp/${exp_tag}
-decode_dir: ${exp_dir}/decode
+infer_dir: ${exp_dir}/infer
 
 # Dataset definition for test splits only.
 dataset:
@@ -25,6 +25,10 @@ model:
   asr_train_config: ${exp_dir}/config.yaml
   asr_model_file: ${exp_dir}/last.ckpt
 
+# Inference I/O.
+input_key: speech
+output_fn: src.infer.output_fn
+
 # Parallel execution used during data loading / scoring (defaults to local).
 parallel:
   env: local

diff --git a/egs3/TEMPLATE/asr/conf/measure.yaml b/egs3/TEMPLATE/asr/conf/measure.yaml
@@ -8,6 +8,7 @@ exp_tag: asr_template_eval
 exp_dir: ${recipe_dir}/exp/${exp_tag}
 stats_dir: ${recipe_dir}/exp/stats
 decode_dir: ${exp_dir}/decode
+infer_dir: ${exp_dir}/infer
 dataset_dir: /path/to/your/dataset
 
 # Dataset definition for test splits only (used to enumerate test names).

diff --git a/egs3/mini_an4/asr/conf/infer.yaml b/egs3/mini_an4/asr/conf/infer.yaml
@@ -4,7 +4,7 @@ recipe_dir: .
 data_dir: ${recipe_dir}/data
 exp_tag: ${load_yaml:conf/train.yaml,exp_tag}
 exp_dir: ${recipe_dir}/exp/${exp_tag}
-decode_dir: ${exp_dir}/decode
+infer_dir: ${exp_dir}/infer
 dataset_dir: ${data_dir}/mini_an4
 
 dataset:
@@ -25,3 +25,6 @@ model:
   asr_model_file: ${exp_dir}/last.ckpt
   beam_size: 1
   ctc_weight: 0.3
+
+input_key: speech
+output_fn: src.infer.output_fn
diff --git a/egs3/mini_an4/asr/conf/infer_transducer.yaml b/egs3/mini_an4/asr/conf/infer_transducer.yaml
@@ -4,7 +4,7 @@ recipe_dir: .
 data_dir: ${recipe_dir}/data
 exp_tag: ${load_yaml:conf/train.yaml,exp_tag}
 exp_dir: ${recipe_dir}/exp/${exp_tag}
-decode_dir: ${exp_dir}/decode
+infer_dir: ${exp_dir}/infer
 dataset_dir: ${data_dir}/mini_an4
 
 dataset:
@@ -24,3 +24,6 @@ model:
   asr_train_config: ${exp_dir}/config.yaml
   asr_model_file: ${exp_dir}/last.ckpt
   beam_size: 1
+
+input_key: speech
+output_fn: src.infer.output_fn_transducer
diff --git a/egs3/mini_an4/asr/conf/measure.yaml b/egs3/mini_an4/asr/conf/measure.yaml
@@ -6,6 +6,7 @@ exp_tag: ${load_yaml:conf/train.yaml,exp_tag}
 exp_dir: ${recipe_dir}/exp/${exp_tag}
 stats_dir: ${recipe_dir}/exp/stats
 decode_dir: ${exp_dir}/decode
+infer_dir: ${exp_dir}/infer
 dataset_dir: ${data_dir}/mini_an4
 
 dataset:

diff --git a/egs3/mini_an4/asr/src/infer.py b/egs3/mini_an4/asr/src/infer.py
@@ -0,0 +1,17 @@
+"""Inference output helpers for ASR recipes."""
+
+
+def output_fn(*, data, model_output, idx):
+    """Build a dict of outputs for SCP writing."""
+    uttid = data.get("uttid", str(idx))
+    hyp = model_output[0][0]
+    ref = data.get("text", "")
+    return {"uttid": uttid, "hyp": hyp, "ref": ref}
+
+
+def output_fn_transducer(*, data, model_output, idx):
+    """Build a dict of outputs for transducer models."""
+    uttid = data.get("uttid", str(idx))
+    hyp = model_output[0]
+    ref = data.get("text", "")
+    return {"uttid": uttid, "hyp": hyp, "ref": ref}