Zarnabh commited on
Commit
193787f
·
verified ·
1 Parent(s): bfe39fc

End of training

Browse files
Files changed (5) hide show
  1. README.md +8 -8
  2. config.json +2 -2
  3. generation_config.json +4 -2
  4. model.safetensors +1 -1
  5. training_args.bin +2 -2
README.md CHANGED
@@ -20,7 +20,7 @@ should probably proofread and complete it, then remove this comment. -->
20
 
21
  This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the Common Voice 20.0 dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 0.7531
24
 
25
  ## Model description
26
 
@@ -51,15 +51,15 @@ The following hyperparameters were used during training:
51
 
52
  | Training Loss | Epoch | Step | Validation Loss |
53
  |:-------------:|:------:|:----:|:---------------:|
54
- | 0.2301 | 0.7418 | 500 | 0.7900 |
55
- | 0.1639 | 1.4837 | 1000 | 0.8025 |
56
- | 0.094 | 2.2255 | 1500 | 0.7681 |
57
- | 0.0315 | 2.9674 | 2000 | 0.7531 |
58
 
59
 
60
  ### Framework versions
61
 
62
- - Transformers 4.55.1
63
  - Pytorch 2.7.0+cu126
64
- - Datasets 4.0.0
65
- - Tokenizers 0.21.4
 
20
 
21
  This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the Common Voice 20.0 dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 0.6051
24
 
25
  ## Model description
26
 
 
51
 
52
  | Training Loss | Epoch | Step | Validation Loss |
53
  |:-------------:|:------:|:----:|:---------------:|
54
+ | 1.1012 | 0.7418 | 500 | 0.7840 |
55
+ | 0.5201 | 1.4837 | 1000 | 0.6686 |
56
+ | 0.3054 | 2.2255 | 1500 | 0.6400 |
57
+ | 0.1285 | 2.9674 | 2000 | 0.6051 |
58
 
59
 
60
  ### Framework versions
61
 
62
+ - Transformers 4.56.2
63
  - Pytorch 2.7.0+cu126
64
+ - Datasets 4.1.1
65
+ - Tokenizers 0.22.0
config.json CHANGED
@@ -16,6 +16,7 @@
16
  "decoder_layers": 12,
17
  "decoder_start_token_id": 50258,
18
  "dropout": 0.0,
 
19
  "encoder_attention_heads": 12,
20
  "encoder_ffn_dim": 3072,
21
  "encoder_layerdrop": 0.0,
@@ -52,8 +53,7 @@
52
  "num_mel_bins": 80,
53
  "pad_token_id": 50257,
54
  "scale_embedding": false,
55
- "torch_dtype": "float32",
56
- "transformers_version": "4.55.1",
57
  "use_cache": true,
58
  "use_weighted_layer_sum": false,
59
  "vocab_size": 51865
 
16
  "decoder_layers": 12,
17
  "decoder_start_token_id": 50258,
18
  "dropout": 0.0,
19
+ "dtype": "float32",
20
  "encoder_attention_heads": 12,
21
  "encoder_ffn_dim": 3072,
22
  "encoder_layerdrop": 0.0,
 
53
  "num_mel_bins": 80,
54
  "pad_token_id": 50257,
55
  "scale_embedding": false,
56
+ "transformers_version": "4.56.2",
 
57
  "use_cache": true,
58
  "use_weighted_layer_sum": false,
59
  "vocab_size": 51865
generation_config.json CHANGED
@@ -47,7 +47,9 @@
47
  ],
48
  "bos_token_id": 50257,
49
  "decoder_start_token_id": 50258,
50
- "eos_token_id": 50257,
 
 
51
  "forced_decoder_ids": null,
52
  "is_multilingual": true,
53
  "lang_to_id": {
@@ -251,5 +253,5 @@
251
  "transcribe": 50359,
252
  "translate": 50358
253
  },
254
- "transformers_version": "4.55.1"
255
  }
 
47
  ],
48
  "bos_token_id": 50257,
49
  "decoder_start_token_id": 50258,
50
+ "eos_token_id": [
51
+ 50257
52
+ ],
53
  "forced_decoder_ids": null,
54
  "is_multilingual": true,
55
  "lang_to_id": {
 
253
  "transcribe": 50359,
254
  "translate": 50358
255
  },
256
+ "transformers_version": "4.56.2"
257
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6f7f23f7012e6a0282d4a42cde1de14bd1c166e582a1b4328b2afd8bfbacc927
3
  size 966995080
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:20c72f79f09b181ad21fbe2c1434c032376dc083f0beb9663b08cc2441bb98a5
3
  size 966995080
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:24029a28caf083de2964d2390e99a5de3b26e612301689ddc0ec2a6b467d479f
3
- size 5841
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a0a5392fbd6d7d272e54fe9b56130db6a6cfaaec42b8e231024748589cc2e27b
3
+ size 5905