Update README.md
Browse files
README.md
CHANGED
|
@@ -50,6 +50,8 @@ print(generated_text)
|
|
| 50 |
|
| 51 |
### Training Hyperparameters
|
| 52 |
|
|
|
|
|
|
|
| 53 |
|predict_with_generate|True|
|
| 54 |
|evaluation_strategy|"steps"|
|
| 55 |
|per_device_train_batch_size|32|
|
|
@@ -57,8 +59,5 @@ print(generated_text)
|
|
| 57 |
|num_train_epochs|2|
|
| 58 |
|fp16|True|
|
| 59 |
|learning_rate|4e-5|
|
| 60 |
-
|output_dir|"./models"|
|
| 61 |
-
|save_steps|2000|
|
| 62 |
-
|eval_steps|1000|
|
| 63 |
|warmup_steps|2000|
|
| 64 |
|weight_decay|0.01|
|
|
|
|
| 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|
|
|
|
|
| 59 |
|num_train_epochs|2|
|
| 60 |
|fp16|True|
|
| 61 |
|learning_rate|4e-5|
|
|
|
|
|
|
|
|
|
|
| 62 |
|warmup_steps|2000|
|
| 63 |
|weight_decay|0.01|
|