Skip to content

V
VIVAS Viet OCR
Commit e61ef9b9 authored by Cao Duc Anh's avatar Cao Duc Anh
Browse files
Options

edit readme

parent 9a3a4aa1
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
README.md
View file @ e61ef9b9
...@@ -17,8 +17,7 @@ images/20140603_0003_BCCTC_tg_12.png chăng đó là kết quả hoàn hảo tro ...@@ -17,8 +17,7 @@ images/20140603_0003_BCCTC_tg_12.png chăng đó là kết quả hoàn hảo tro
``` ```
### 2. Config ### 2. Config
Chỉnh sửa file config.yml nếu cần. <br> Chỉnh sửa file **config.yml**. Các thông tin cần lưu ý:
Các thông tin cần lưu ý:
``` ```
dataset: dataset:
image_height: 32 image_height: 32
...@@ -34,7 +33,8 @@ trainer: ...@@ -34,7 +33,8 @@ trainer:
print_every: 20 print_every: 20
valid_every: 20 valid_every: 20
``` ```
metrics: số mẫu được dùng để valid metrics: số mẫu được dùng để tính acc full seq và acc per char, <= len(test_annotation). <br>
test_annotation dùng để tính valid loss.
### 3. Run training ### 3. Run training
``` ```
......
config.yml
View file @ e61ef9b9
...@@ -48,7 +48,7 @@ pretrain: https://vocr.vn/data/vietocr/vgg_transformer.pth ...@@ -48,7 +48,7 @@ pretrain: https://vocr.vn/data/vietocr/vgg_transformer.pth
quiet: false quiet: false
seq_modeling: transformer seq_modeling: transformer
trainer: trainer:
batch_size: 1 batch_size: 8
checkpoint: ./checkpoint/transformerocr_checkpoint.pth checkpoint: ./checkpoint/transformerocr_checkpoint.pth
export: ./weights/transformerocr.pth export: ./weights/transformerocr.pth
iters: 20000 iters: 20000
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment