Update README.md
Browse files
README.md
CHANGED
|
@@ -45,21 +45,20 @@ print(generated_text)
|
|
| 45 |
### Training Data
|
| 46 |
|
| 47 |
100k words generated by TextRecognitionDataGenerator(trdg) : https://github.com/Belval/TextRecognitionDataGenerator/blob/master/trdg/run.py
|
|
|
|
| 48 |
120k words from AI-hub OCR words dataset : https://aihub.or.kr/aihubdata/data/view.do?currMenu=115&topMenu=100&dataSetSn=81
|
| 49 |
|
| 50 |
### Training Hyperparameters
|
| 51 |
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
weight_decay=0.01
|
| 65 |
-
)
|
|
|
|
| 45 |
### Training Data
|
| 46 |
|
| 47 |
100k words generated by TextRecognitionDataGenerator(trdg) : https://github.com/Belval/TextRecognitionDataGenerator/blob/master/trdg/run.py
|
| 48 |
+
|
| 49 |
120k words from AI-hub OCR words dataset : https://aihub.or.kr/aihubdata/data/view.do?currMenu=115&topMenu=100&dataSetSn=81
|
| 50 |
|
| 51 |
### Training Hyperparameters
|
| 52 |
|
| 53 |
+
|predict_with_generate|True|
|
| 54 |
+
|evaluation_strategy|"steps"|
|
| 55 |
+
|per_device_train_batch_size|32|
|
| 56 |
+
|per_device_eval_batch_size|32|
|
| 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|
|
|
|
|
|
|