Update README.md

README.md

@@ -34,6 +34,11 @@ python3 ./trdg/run.py -i ocr_dataset_poc.txt -w 5 -t {num_cores} -f 64 -l ko -c
 ```
 
 ## Training
+
+### Base model
+The encoder model used `facebook/deit-base-distilled-patch16-384` and the decoder model used `klue/roberta-base`. It is easier than training by starting weights from `microsoft/trocr-base-stage1`.
+
+### Parameters
 We used heuristic parameters without separate hyperparameter tuning.
 - learning_rate = 4e-5
 - epochs = 25