snehalyelmati commited on
Commit
d5a2c3c
1 Parent(s): a9d85f6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -1
README.md CHANGED
@@ -14,4 +14,14 @@ Based on the "google/mt5-small" pre-trained model. Fine-tuned it on Hindi to Eng
14
  - batch size = 16
15
  - learning rate = 5e-4
16
  - number of batches = int(np.ceil(len(dataset) / batch size))
17
- - n_warmup_steps = int(number of epochs * number of batches * 0.01)
 
 
 
 
 
 
 
 
 
 
 
14
  - batch size = 16
15
  - learning rate = 5e-4
16
  - number of batches = int(np.ceil(len(dataset) / batch size))
17
+ - n_warmup_steps = int(number of epochs * number of batches * 0.01)
18
+
19
+ ### Training Loss
20
+
21
+ ![loss.png](./h2e_loss.png)
22
+
23
+ ### Examples
24
+
25
+ ![example_1.png](./h2e_example1.png)
26
+ ![example_2.png](./h2e_example2.png)
27
+ ![example_3.png](./h2e_example3.png)