Skip to content

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

edit config

parent a9933f90
Hide whitespace changes
Inline Side-by-side
Showing with 10 additions and 12 deletions
config.yml
View file @ 59e98787
...@@ -48,22 +48,19 @@ pretrain: https://vocr.vn/data/vietocr/vgg_seq2seq.pth ...@@ -48,22 +48,19 @@ pretrain: https://vocr.vn/data/vietocr/vgg_seq2seq.pth
quiet: false quiet: false
seq_modeling: seq2seq seq_modeling: seq2seq
trainer: trainer:
batch_size: 8 batch_size: 12
checkpoint: ./checkpoint/transformerocr_checkpoint.pth checkpoint: ./checkpoint/transformerocr_checkpoint.pth
export: ./weights/vetocr.pth export: ./weights/vetocr.pth
iters: 20000 iters: 2000
log: ./train.log log: ./train.log
metrics: 2 metrics: 200
print_every: 20 print_every: 20
valid_every: 20 valid_every: 20
transformer: transformer:
d_model: 256 encoder_hidden: 256
dim_feedforward: 2048 decoder_hidden: 256
max_seq_length: 1024 img_channel: 256
nhead: 8 decoder_embedded: 256
num_decoder_layers: 6 dropout: 0.1
num_encoder_layers: 6
pos_dropout: 0.1
trans_dropout: 0.1
vocab: 'aAàÀảẢãÃáÁạẠăĂằẰẳẲẵẴắẮặẶâÂầẦẩẨẫẪấẤậẬbBcCdDđĐeEèÈẻẺẽẼéÉẹẸêÊềỀểỂễỄếẾệỆfFgGhHiIìÌỉỈĩĨíÍịỊjJkKlLmMnNoOòÒỏỎõÕóÓọỌôÔồỒổỔỗỖốỐộỘơƠờỜởỞỡỠớỚợỢpPqQrRsStTuUùÙủỦũŨúÚụỤưƯừỪửỬữỮứỨựỰvVwWxXyYỳỲỷỶỹỸýÝỵỴzZ0123456789!"#$%&''()*+,-./:;<=>?@[\]^_`{|}~ ' vocab: 'aAàÀảẢãÃáÁạẠăĂằẰẳẲẵẴắẮặẶâÂầẦẩẨẫẪấẤậẬbBcCdDđĐeEèÈẻẺẽẼéÉẹẸêÊềỀểỂễỄếẾệỆfFgGhHiIìÌỉỈĩĨíÍịỊjJkKlLmMnNoOòÒỏỎõÕóÓọỌôÔồỒổỔỗỖốỐộỘơƠờỜởỞỡỠớỚợỢpPqQrRsStTuUùÙủỦũŨúÚụỤưƯừỪửỬữỮứỨựỰvVwWxXyYỳỲỷỶỹỸýÝỵỴzZ0123456789!"#$%&''()*+,-./:;<=>?@[\]^_`{|}~ '
weights: https://vocr.vn/data/vietocr/vgg_seq2seq.pth weights: https://vocr.vn/data/vietocr/vgg_seq2seq.pth
\ No newline at end of file
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