End of training

README.md

@@ -1,4 +1,5 @@
 ---
+license: mit
 tags:
 - generated_from_trainer
 datasets:
@@ -13,7 +14,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # donut_pdf_ocr
 
-This model was trained from scratch on the imagefolder dataset.
+This model is a fine-tuned version of [naver-clova-ix/donut-base](https://huggingface.co/naver-clova-ix/donut-base) on the imagefolder dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.0821
 
 ## Model description
 
@@ -34,7 +37,7 @@ More information needed
 The following hyperparameters were used during training:
 - learning_rate: 2e-05
 - train_batch_size: 2
-- eval_batch_size: 8
+- eval_batch_size: 2
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
@@ -43,6 +46,38 @@ The following hyperparameters were used during training:
 
 ### Training results
 
+| Training Loss | Epoch | Step | Validation Loss |
+|:-------------:|:-----:|:----:|:---------------:|
+| 0.6648        | 1.0   | 47   | 0.2550          |
+| 0.3772        | 2.0   | 94   | 0.0895          |
+| 0.1732        | 3.0   | 141  | 0.0862          |
+| 0.1244        | 4.0   | 188  | 0.0718          |
+| 0.0867        | 5.0   | 235  | 0.0473          |
+| 0.1035        | 6.0   | 282  | 0.0640          |
+| 0.1703        | 7.0   | 329  | 0.0877          |
+| 0.0324        | 8.0   | 376  | 0.0722          |
+| 0.033         | 9.0   | 423  | 0.0701          |
+| 0.0404        | 10.0  | 470  | 0.0627          |
+| 0.0198        | 11.0  | 517  | 0.0770          |
+| 0.007         | 12.0  | 564  | 0.0730          |
+| 0.0207        | 13.0  | 611  | 0.0885          |
+| 0.0528        | 14.0  | 658  | 0.0898          |
+| 0.0189        | 15.0  | 705  | 0.0588          |
+| 0.0977        | 16.0  | 752  | 0.0563          |
+| 0.0542        | 17.0  | 799  | 0.0647          |
+| 0.0114        | 18.0  | 846  | 0.0674          |
+| 0.0056        | 19.0  | 893  | 0.0726          |
+| 0.005         | 20.0  | 940  | 0.0797          |
+| 0.0005        | 21.0  | 987  | 0.0704          |
+| 0.0023        | 22.0  | 1034 | 0.0864          |
+| 0.0105        | 23.0  | 1081 | 0.0830          |
+| 0.0014        | 24.0  | 1128 | 0.0806          |
+| 0.0006        | 25.0  | 1175 | 0.0826          |
+| 0.0012        | 26.0  | 1222 | 0.0790          |
+| 0.0006        | 27.0  | 1269 | 0.0751          |
+| 0.0022        | 28.0  | 1316 | 0.0786          |
+| 0.0075        | 29.0  | 1363 | 0.0816          |
+| 0.0024        | 30.0  | 1410 | 0.0821          |
 
 
 ### Framework versions

runs/Apr21_10-20-11_1945e9cc3f65/events.out.tfevents.1682072420.1945e9cc3f65.310.2

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d0e404f0713e08c4c43ea803688a005ed9b157b8d97ac449b70c91d25b56c5cc
-size 27656
+oid sha256:eb02eef699e246acdd5983e263e4ba81eadf6ffb66e439fa7dba8e8dec10e0c7
+size 28010

special_tokens_map.json

@@ -1,5 +1,17 @@
 {
   "additional_special_tokens": [
+    "<s_patient_name>",
+    "</s_patient_name>",
+    "<s_patient_id>",
+    "</s_patient_id>",
+    "<s_gender>",
+    "</s_gender>",
+    "<s_discharge_diagnosis>",
+    "</s_discharge_diagnosis>",
+    "<s_discharge_date>",
+    "</s_discharge_date>",
+    "<s_admit_date>",
+    "</s_admit_date>",
     "<s>",
     "</s>"
   ],