-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathrun_decode.sh
More file actions
20 lines (20 loc) · 889 Bytes
/
run_decode.sh
File metadata and controls
20 lines (20 loc) · 889 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
#!/bin/bash
checkpoint_name=hc_ctc_all_then_swb_H512L24DS8E15
test_path=/research/nfs_fosler_1/vishal/text/hub5_2000/2000_hub5_eng_eval_tr/reference/english/switchboard.csv
PTH=decodes/${checkpoint_name}_switchboard
world_size=4
rm -vrf $PTH/*.log
rm -vrf $PTH/*.txt
#python main.py \
#OMP_NUM_THREADS=8 python -m torch.distributed.launch --nproc_per_node=$world_size --master_port=33543 decode.py \
OMP_NUM_THREADS=8 torchrun --standalone --nnodes=1 --nproc_per_node=$world_size decode.py \
--config-path="configs/" \
--config-name="hc_ctc_all_then_swb" \
corpus='switchboard' \
distributed.world_size=$world_size \
paths.test_path="${test_path}" \
paths.ckpt_path="/research/nfs_fosler_1/vishal/saved_models/${checkpoint_name}.pth.tar" \
paths.decode_path="${PTH}"
rm -vrf $PTH/full.txt
cat ${PTH}/{0..3}.txt > "${PTH}/full.txt"
python evaluate.py --path "${PTH}/full.txt"