Update README.md
Browse files
README.md
CHANGED
|
@@ -50,14 +50,14 @@ print(generated_text)
|
|
| 50 |
|
| 51 |
### Training Hyperparameters
|
| 52 |
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
|
|
|
| 50 |
|
| 51 |
### Training Hyperparameters
|
| 52 |
|
| 53 |
+
|hyperparameters|values|
|
| 54 |
+
|-----------------------------|-------|
|
| 55 |
+
|predict_with_generate|True|
|
| 56 |
+
|evaluation_strategy|"steps"|
|
| 57 |
+
|per_device_train_batch_size|32|
|
| 58 |
+
|per_device_eval_batch_size|32|
|
| 59 |
+
|num_train_epochs|2|
|
| 60 |
+
|fp16|True|
|
| 61 |
+
|learning_rate|4e-5|
|
| 62 |
+
|warmup_steps|2000|
|
| 63 |
+
|weight_decay|0.01|
|