zhifu gao
2024-04-17 e8f80e96f99cb856423d030c7d055c302a6d3278
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"
inference_checkpoint="model.pt.avg10"
inference_scp="wav.scp"
inference_batch_size=32
inference_batch_size=1
# data
raw_data=../raw_data