zhifu gao
2024-04-02 48693b45c021a842ea964c9dc99479b61eac062f
examples/aishell/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