update model card README.md
Browse files
README.md
CHANGED
@@ -52,20 +52,21 @@ More information needed
|
|
52 |
### Training hyperparameters
|
53 |
|
54 |
The following hyperparameters were used during training:
|
55 |
-
- learning_rate:
|
56 |
-
- train_batch_size:
|
57 |
-
- eval_batch_size:
|
58 |
- seed: 42
|
59 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
60 |
- lr_scheduler_type: linear
|
61 |
-
- num_epochs:
|
62 |
|
63 |
### Training results
|
64 |
|
65 |
-
| Training Loss | Epoch | Step | Validation Loss | Bleu
|
66 |
-
|
67 |
-
|
|
68 |
-
|
|
|
|
69 |
|
70 |
|
71 |
### Framework versions
|
|
|
52 |
### Training hyperparameters
|
53 |
|
54 |
The following hyperparameters were used during training:
|
55 |
+
- learning_rate: 0.001
|
56 |
+
- train_batch_size: 16
|
57 |
+
- eval_batch_size: 16
|
58 |
- seed: 42
|
59 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
60 |
- lr_scheduler_type: linear
|
61 |
+
- num_epochs: 3
|
62 |
|
63 |
### Training results
|
64 |
|
65 |
+
| Training Loss | Epoch | Step | Validation Loss | Bleu | Gen Len |
|
66 |
+
|:-------------:|:-----:|:----:|:---------------:|:------:|:-------:|
|
67 |
+
| No log | 1.0 | 250 | nan | 2.9676 | 12.146 |
|
68 |
+
| 2.5985 | 2.0 | 500 | nan | 0.0 | 0.0 |
|
69 |
+
| 2.5985 | 3.0 | 750 | nan | 0.0 | 0.0 |
|
70 |
|
71 |
|
72 |
### Framework versions
|