维石
2024-04-17 149063ced4d2d5269f0472677228eadfcb4a4d8a
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.avg5"
inference_checkpoint="model.pt.avg10"
inference_scp="wav.scp"
inference_batch_size=32
inference_batch_size=1
# data
raw_data=../raw_data