edit config

config.yml

@@ -48,22 +48,19 @@ pretrain: https://vocr.vn/data/vietocr/vgg_seq2seq.pth
 quiet: false
 seq_modeling: seq2seq
 trainer:
-  batch_size: 8
+  batch_size: 12
   checkpoint: ./checkpoint/transformerocr_checkpoint.pth
   export: ./weights/vetocr.pth
-  iters: 20000
+  iters: 2000
   log: ./train.log
-  metrics: 2
+  metrics: 200
   print_every: 20
   valid_every: 20
 transformer:
-  d_model: 256
-  dim_feedforward: 2048
-  max_seq_length: 1024
-  nhead: 8
-  num_decoder_layers: 6
-  num_encoder_layers: 6
-  pos_dropout: 0.1
-  trans_dropout: 0.1
+  encoder_hidden: 256
+  decoder_hidden: 256
+  img_channel: 256
+  decoder_embedded: 256
+  dropout: 0.1
 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