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|
|
|
|
|