01Yassine commited on
Commit
dca60a7
·
verified ·
1 Parent(s): 5a7de6d

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: PartialSpoof_wavlm_TCM_NoAug_seed2
2
+ output_dir: ./outputs/batch7_PartialSpoof/
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/PartialSpoof/train.parquet
13
+ dataset_names:
14
+ - PartialSpoof_Train
15
+ batch_size: 16
16
+ shuffle: true
17
+ num_workers: 4
18
+ augment_transform: []
19
+ val:
20
+ dataset_type: StandardDataset
21
+ parquet_files:
22
+ - /netscratch/yelkheir/DeepFense/DeepFense/deepfense/config/parquets/PartialSpoof/dev.parquet
23
+ dataset_names:
24
+ - PartialSpoof_Val
25
+ batch_size: 1
26
+ shuffle: false
27
+ num_workers: 4
28
+ test:
29
+ dataset_type: StandardDataset
30
+ parquet_files:
31
+ - /netscratch/yelkheir/DeepFense/DeepFense/deepfense/config/parquets/PartialSpoof/eval.parquet
32
+ dataset_names:
33
+ - PartialSpoof_Eval
34
+ batch_size: 1
35
+ shuffle: false
36
+ num_workers: 4
37
+ model:
38
+ type: StandardDetector
39
+ frontend:
40
+ type: wavlm
41
+ args:
42
+ ckpt_path: /netscratch/yelkheir/ssl_models/WavLM-Large.pt
43
+ freeze: false
44
+ source: unil
45
+ backend:
46
+ type: TCM
47
+ args:
48
+ input_dim: 1024
49
+ emb_size: 128
50
+ heads: 4
51
+ num_encoders: 1
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/batch7_PartialSpoof/PartialSpoof_wavlm_TCM_NoAug_seed2_20260121_080548