feat: uploaded VTC 2.0 model and config
Browse files- model/best.ckpt +3 -0
- 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
|