File size: 431 Bytes
7bd8b78
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#!/bin/bash

# μ‚¬μš©ν•  GPU 개수 (μ‹œμŠ€ν…œμ˜ λͺ¨λ“  GPU)
export CUDA_VISIBLE_DEVICES=6,7
export NUM_GPUS=2

echo "Starting EMOVA ASR evaluation on ${NUM_GPUS} GPUs..."

torchrun --nproc_per_node=${NUM_GPUS} \
         --master_port=12351 \
         /home/work/AIDAS/EMOVA_speech_tokenizer/eval.py \
    --batch_size=4 \
    --dataset_config="test" \
    --split="clean" \
    --wandb_run_name="emova-7b-librispeech-test-clean"