arxaqapi commited on
Commit
91e67b5
·
verified ·
1 Parent(s): 8e3799f

feat: uploaded VTC 2.0 model and config

Browse files
Files changed (2) hide show
  1. model/best.ckpt +3 -0
  2. model/config.yml +37 -0
model/best.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8438b9a822a3d80fe8bd24385df4c1133ab74dd69cc1f7ad14d89e753a177af7
3
+ size 1099175630
model/config.yml ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ wandb:
2
+ offline: false
3
+ project: babyhubert-finetune
4
+ name: train
5
+ data:
6
+ dataset_path: data/baby_train
7
+ classes:
8
+ - KCHI
9
+ - OCH
10
+ - MAL
11
+ - FEM
12
+ dataset_multiplier: 1.0
13
+ audio:
14
+ chunk_duration_s: 4
15
+ sample_rate: 16000
16
+ strict_frames: false
17
+ model:
18
+ name: surgical_hubert_hydra
19
+ chkp_path: null
20
+ config:
21
+ wav_encoder: ""
22
+ encoder_layers: []
23
+ reduction: weighted
24
+ classifier: 256
25
+ train:
26
+ lr: 1.0e-05
27
+ batch_size: 128
28
+ max_epochs: 3
29
+ validation_metric: loss
30
+ extra_val_metrics:
31
+ - loss
32
+ - f1_score
33
+ profiler: null
34
+ dataloader:
35
+ num_workers: 8
36
+ scheduler:
37
+ patience: 3