Rohan432 commited on
Commit
d9c8dbe
·
verified ·
1 Parent(s): 99c8b44

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -13
README.md CHANGED
@@ -5,24 +5,24 @@ base_model:
5
  pipeline_tag: automatic-speech-recognition
6
  ---
7
 
8
- Description
9
  This model is a fine-tuned version of ishmamzarif/bangla_asr_augmented_bangla-whisper-epoch-11 .
10
  ishmamzarif/bangla_asr_augmented_bangla-whisper-epoch-11 is a finetuned version of bangla-speech-processing/BanglaASR on Bangla speech data.
11
 
12
- Environment:
13
- Python version: 3.12.12
14
- PyTorch version: 2.8.0+cu126
15
- Librosa version: 0.10.1
16
- NumPy version: 1.26.4
17
 
18
- Training Parameters:
19
 
20
- BATCH_SIZE = 4
21
- GRADIENT_ACCUMULATION_STEPS = 4 # Effective batch size = 16
22
- LEARNING_RATE = 2e-5
23
- WARMUP_STEPS = 400
24
- NUM_TRAIN_EPOCHS = 8
25
- LOGGING_STEPS = 50
26
 
27
 
28
  | **Epoch** | **Training Loss** | **Validation Loss** | **WER (%)** | **Normalized Levenshtein Similarity (%)** |
 
5
  pipeline_tag: automatic-speech-recognition
6
  ---
7
 
8
+ ## Description
9
  This model is a fine-tuned version of ishmamzarif/bangla_asr_augmented_bangla-whisper-epoch-11 .
10
  ishmamzarif/bangla_asr_augmented_bangla-whisper-epoch-11 is a finetuned version of bangla-speech-processing/BanglaASR on Bangla speech data.
11
 
12
+ ## Environment:
13
+ - Python version: 3.12.12
14
+ - PyTorch version: 2.8.0+cu126
15
+ - Librosa version: 0.10.1
16
+ - NumPy version: 1.26.4
17
 
18
+ ## Training Parameters:
19
 
20
+ - BATCH_SIZE = 4
21
+ - GRADIENT_ACCUMULATION_STEPS = 4 # Effective batch size = 16
22
+ - LEARNING_RATE = 2e-5
23
+ - WARMUP_STEPS = 400
24
+ - NUM_TRAIN_EPOCHS = 8
25
+ - LOGGING_STEPS = 50
26
 
27
 
28
  | **Epoch** | **Training Loss** | **Validation Loss** | **WER (%)** | **Normalized Levenshtein Similarity (%)** |