MultiConvAdapater_AudioDeepfake/config.py at main · Yaselley/MultiConvAdapater_AudioDeepfake · 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
# paths.py
import os

# ---------------------- ASV2019 ---------------------- #
ASV19 = {
    "train_flac": "/ds/audio/LA_19/ASVspoof2019_LA_train/flac/",
    "dev_flac": "/ds/audio/LA_19/ASVspoof2019_LA_dev/flac/",
    "train_protocol": "./protocols/ASV19/train.txt",
    "dev_protocol": "./protocols/ASV19/dev.txt"
}

# ---------------------- ASV5 ---------------------- #
ASV5 = {
    "train_flac": "/ds-slt/audio/ASVSpoof2024/flac_T/",
    "dev_flac": "/ds-slt/audio/ASVSpoof2024/flac_D/",
    "train_protocol": "./protocols/ASV5/asvspoof5.train.txt",
    "dev_protocol": "./protocols/ASV5/asvspoof5.dev.txt"
}

# ---------------------- ASV5 ---------------------- #
TEST = {
    "flac": "/ds/audio/LA_19/ASVspoof2019_LA_train/flac/",
    "protocol": "./protocols/ASV19/train.txt",
    "ext": "flac"
}

# ---------------------- Default settings ---------------------- #
DEFAULT_MAX_LEN = 64600
SAMPLE_RATE = 16000
W2V_XLSR_CONFIG = "./protocols/w2v_config.json"
W2V_XLSR_CKPT = "./ssl_models/xlsr2_300m.pt"

DEVICE = "cuda"
BATCH_SIZE = 128
NUM_WORKERS = 1
PIN_MEMORY = True
SHUFFLE = False

# ---------------------- Default Output Folder ---------------------- #
DEFAULT_OUT_DIR = "./outputs/"

# ---------------------- BEST EPOCH ---------------------- #
BEST_CKPT = "./protocols/CKPT/ASV19/SSL_best.pt"
BEST_CONFIG = "./protocols/CKPT/ASV19/config.json"