01Yassine's picture
Upload config.yaml with huggingface_hub
1906f65 verified
exp_name: CompSpoof_eat_MLP_NoAug_seed2
output_dir: ./outputs/batch8_CompSpoof/
seed: 2
data:
sampling_rate: 16000
root_dir: /ds-slt/audio/yelkheir/CompSpoof
label_map:
spoof_spoof: 0
bonafide_bonafide: 1
spoof_bonafide: 0
bonafide_spoof: 0
original: 1
train:
dataset_type: StandardDataset
parquet_files:
- /netscratch/yelkheir/DeepFense/DeepFense/deepfense/config/parquets/CompSpoof/CompSpoof_train.parquet
dataset_names:
- CompSpoof_Train
batch_size: 32
shuffle: true
num_workers: 4
base_transform:
- type: pad
max_len: 64000
random_pad: true
pad_type: repeat
augment_transform: []
val:
dataset_type: StandardDataset
parquet_files:
- /netscratch/yelkheir/DeepFense/DeepFense/deepfense/config/parquets/CompSpoof/CompSpoof_dev.parquet
dataset_names:
- CompSpoof_Val
batch_size: 32
shuffle: false
num_workers: 4
base_transform:
- type: pad
max_len: 64000
pad_type: repeat
test:
dataset_type: StandardDataset
parquet_files:
- /netscratch/yelkheir/DeepFense/DeepFense/deepfense/config/parquets/CompSpoof/CompSpoof_eval.parquet
dataset_names:
- CompSpoof_Eval
batch_size: 32
shuffle: false
num_workers: 4
base_transform:
- type: pad
max_len: 64000
pad_type: repeat
model:
type: StandardDetector
frontend:
type: eat
args:
source: huggingface
ckpt_path: worstchan/EAT-large_epoch20_pretrain
freeze: false
backend:
type: MLP
args:
input_dim: 1024
projection:
- 128
pooling_type: mean
loss:
- type: CrossEntropy
embedding_dim: 128
n_classes: 2
weight: 1.0
training:
trainer: StandardTrainer
epochs: 100
early_stopping_patience: 7
optimizer:
type: adam
lr: 1.0e-06
metrics:
ACC: {}
EER: {}
eval_every_epochs: 1
device: cuda
output_dir: ./outputs/batch8_CompSpoof/CompSpoof_eat_MLP_NoAug_seed2_20260129_152218