feat: VTC 2.1

README.md

@@ -3,7 +3,7 @@ base_model:
 - coml/BabyHuBERT
 ---
 
-# VTC2.0
+# VTC 2.1
 
 This repository only contains the model weights.
 For more informations on how to use the model, please look over at the [LAAC-LSCP/VTC](https://github.com/LAAC-LSCP/VTC) repository.

model/best.ckpt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8438b9a822a3d80fe8bd24385df4c1133ab74dd69cc1f7ad14d89e753a177af7
-size 1099175630
+oid sha256:5c3446d037f9c6746cbe19bb77a77973354d3443556f9e2278e83cd4fe7db5a2
+size 1099159455

model/config.toml

@@ -0,0 +1,26 @@
+[logging]
+project = "segma"
+name = "train_VTC-2.1"
+id = "260331-130845"
+offline = false
+tags = ["vtc-2.1", "u20", "cds:10"]
+
+[data]
+dataset_path = ""
+labels = ["KCHI", "OCH", "MAL", "FEM"]
+dataset_multiplier = 1
+
+[audio]
+chunk_duration_s = 10
+sample_rate = 16_000
+strict_frames = true
+
+[model]
+
+[train]
+learning_rate = 1.0e-05
+batch_size = 128
+max_epochs = 2
+dataload_num_workers = 24
+scheduler_patience = 3
+seed = 9

model/config.yml

@@ -1,37 +0,0 @@
-wandb:
-  offline: false
-  project: babyhubert-finetune
-  name: train
-data:
-  dataset_path: data/baby_train
-  classes:
-  - KCHI
-  - OCH
-  - MAL
-  - FEM
-  dataset_multiplier: 1.0
-audio:
-  chunk_duration_s: 4
-  sample_rate: 16000
-  strict_frames: false
-model:
-  name: surgical_hubert_hydra
-  chkp_path: null
-  config:
-    wav_encoder: ""
-    encoder_layers: []
-    reduction: weighted
-    classifier: 256
-train:
-  lr: 1.0e-05
-  batch_size: 128
-  max_epochs: 3
-  validation_metric: loss
-  extra_val_metrics:
-  - loss
-  - f1_score
-  profiler: null
-  dataloader:
-    num_workers: 8
-  scheduler:
-    patience: 3