Codecs_reconstruction_eval/run_usage.sh at main · shwj114514/Codecs_reconstruction_eval · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
export http_proxy='http://10.2.254.3:8888'
export https_proxy='http://10.2.254.3:8888'

export CUDA_VISIBLE_DEVICES="3"

DEVICE="cuda:0"

RECON_OUTPUT_FOLDER="recon_rebuttal_all"
EXP_FOLDER="exp_recon"

run_model() {
    local MODEL=$1
    local N_QUANTIZERS=$2

    AUDIO_FOLDER="${EXP_FOLDER}/test-clean"
    # AUDIO_FOLDER="${EXP_FOLDER}/tmp"


    echo "Cal entropy model: $MODEL with $N_QUANTIZERS quantizers"

    python metrics/compute_usage.py \
        --device $DEVICE \
        --folder "$AUDIO_FOLDER" \
        --model $MODEL \
        --n_quantizers $N_QUANTIZERS
}

# run_model "Xcodec2" 0
# run_model "Mimi" 6
# run_model "SpeachTokenzier" 1
# run_model "SpeachTokenzier" 2
# run_model "DAC_16k" 2
# run_model "DAC_24k" 1
# run_model "Encodec" 1

# run_model "WavTokenizer600" 0
# run_model "WavTokenizer320" 0
# run_model "BigCodec" 0

# run_model "StableCodec" 0
# run_model "StableCodec_base" 0

# run_model "StableCodec_base_400bps" 0
# run_model "StableCodec_base_700bps" 0
# run_model "StableCodec_base_1000bps" 0


# run_model "Xcodec" 1
# run_model "Xcodec" 2
# run_model "SemanticCsodec_700bps" 0


# run_model "Encodec" 8
# run_model "DAC_24k" 8

# run_model "DAC_16k" 1
# run_model "DAC_16k" 12

# nohup bash run_usage.sh > run_usage.txt 2>&1 &

# run_model "Encodec" 2
# run_model "DAC_16k" 2
run_model "WavTokenizer600" 0