gg4ever commited on
Commit
1fd6fd4
·
1 Parent(s): 91cf8be

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -14
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
- training_args = Seq2SeqTrainingArguments(
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
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|