01Yassine commited on
Commit
4d7141e
·
verified ·
1 Parent(s): 1336da0

Upload config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.yaml +69 -0
config.yaml ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ exp_name: EnvSSD_wav2vec2_MLP_NoAug_seed2
2
+ output_dir: ./outputs/batch5_EnvSSD/
3
+ seed: 2
4
+ data:
5
+ sampling_rate: 16000
6
+ label_map:
7
+ bonafide: 1
8
+ spoof: 0
9
+ train:
10
+ dataset_type: StandardDataset
11
+ parquet_files:
12
+ - /netscratch/yelkheir/DeepFense/DeepFense/deepfense/config/parquets/EnvSSD/envssd_train.parquet
13
+ dataset_names:
14
+ - EnvSSD_Train
15
+ batch_size: 32
16
+ shuffle: true
17
+ num_workers: 4
18
+ base_transform:
19
+ - type: pad
20
+ max_len: 64000
21
+ random_pad: true
22
+ pad_type: repeat
23
+ augment_transform: []
24
+ val:
25
+ dataset_type: StandardDataset
26
+ parquet_files:
27
+ - /netscratch/yelkheir/DeepFense/DeepFense/deepfense/config/parquets/EnvSSD/envssd_val.parquet
28
+ dataset_names:
29
+ - EnvSSD_Val
30
+ batch_size: 32
31
+ shuffle: false
32
+ num_workers: 4
33
+ base_transform:
34
+ - type: pad
35
+ max_len: 64000
36
+ pad_type: repeat
37
+ model:
38
+ type: StandardDetector
39
+ frontend:
40
+ type: wav2vec2
41
+ args:
42
+ ckpt_path: /netscratch/yelkheir/ssl_models/xlsr2_300m.pt
43
+ freeze: false
44
+ source: fairseq
45
+ backend:
46
+ type: MLP
47
+ args:
48
+ input_dim: 1024
49
+ projection:
50
+ - 128
51
+ pooling_type: mean
52
+ loss:
53
+ - type: CrossEntropy
54
+ embedding_dim: 128
55
+ n_classes: 2
56
+ weight: 1.0
57
+ training:
58
+ trainer: StandardTrainer
59
+ epochs: 100
60
+ early_stopping_patience: 7
61
+ optimizer:
62
+ type: adam
63
+ lr: 1.0e-06
64
+ metrics:
65
+ ACC: {}
66
+ EER: {}
67
+ eval_every_epochs: 1
68
+ device: cuda
69
+ output_dir: ./outputs/batch5_EnvSSD/EnvSSD_wav2vec2_MLP_NoAug_seed2_20251227_122732