zhifu gao
2024-04-10 112c8e6eb7c7b5dd150787848a5c1d6111688ed7
examples/aishell/e_branchformer/run.sh
@@ -1,7 +1,7 @@
#!/usr/bin/env bash
CUDA_VISIBLE_DEVICES="0,1"
CUDA_VISIBLE_DEVICES="0,1,2,3"
# general configuration
feats_dir="../DATA" #feature output dictionary
@@ -15,9 +15,9 @@
nj=32
inference_device="cuda" #"cpu"
inference_checkpoint="model.pt.avg5"
inference_checkpoint="model.pt.avg10"
inference_scp="wav.scp"
inference_batch_size=32
inference_batch_size=1
# data
raw_data=../raw_data