Training in progress, epoch 1

config.json

@@ -1,7 +1,7 @@
 {
-  "_name_or_path": "ntu-spml/distilhubert",
+  "_name_or_path": "facebook/hubert-base-ls960",
   "activation_dropout": 0.1,
-  "apply_spec_augment": false,
+  "apply_spec_augment": true,
   "architectures": [
     "HubertForSequenceClassification"
   ],
@@ -42,12 +42,15 @@
   "do_stable_layer_norm": false,
   "eos_token_id": 2,
   "feat_extract_activation": "gelu",
+  "feat_extract_dropout": 0.0,
   "feat_extract_norm": "group",
-  "feat_proj_dropout": 0.0,
-  "feat_proj_layer_norm": false,
-  "final_dropout": 0.0,
+  "feat_proj_dropout": 0.1,
+  "feat_proj_layer_norm": true,
+  "final_dropout": 0.1,
+  "gradient_checkpointing": false,
   "hidden_act": "gelu",
   "hidden_dropout": 0.1,
+  "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label": {
     "0": "01_gunshot",
@@ -74,7 +77,7 @@
     "10_music": "8"
   },
   "layer_norm_eps": 1e-05,
-  "layerdrop": 0.0,
+  "layerdrop": 0.1,
   "mask_feature_length": 10,
   "mask_feature_min_masks": 0,
   "mask_feature_prob": 0.0,
@@ -86,8 +89,9 @@
   "num_conv_pos_embedding_groups": 16,
   "num_conv_pos_embeddings": 128,
   "num_feat_extract_layers": 7,
-  "num_hidden_layers": 2,
+  "num_hidden_layers": 12,
   "pad_token_id": 0,
+  "tokenizer_class": "Wav2Vec2CTCTokenizer",
   "torch_dtype": "float32",
   "transformers_version": "4.48.3",
   "use_weighted_layer_sum": false,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:406b8d132d9eb7b47f819eed4d5b5eae56167f49f8e102b59dbfa5e7692f9b00
-size 94770700
+oid sha256:d4abccd64e1af3e5045c26346b8bdee179518a969ecbdceec7a77c20d3bf9969
+size 378309148

runs/Mar11_13-58-32_2dae0ebb4030/events.out.tfevents.1741701692.2dae0ebb4030.706.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7d6e323afcd8768e5cd993ddde10e8dacc5a37810086b3a1810e731652ef8602
+size 7885

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bb24a8d958f3db0ee175d9b127ce7d32d2a80bfd30d923360a2b539624387930
+oid sha256:d4874e4b9f76d5abe82839aac6f28106b92791c09b816f966a9b1cf81f13d776
 size 5368