hantech commited on
Commit
1787b5b
โ€ข
1 Parent(s): 459b18f

Delete vgg-seq2seq.yaml

Browse files
Files changed (1) hide show
  1. vgg-seq2seq.yaml +0 -90
vgg-seq2seq.yaml DELETED
@@ -1,90 +0,0 @@
1
- project: vietocr_new
2
- name: Train
3
-
4
- device: cuda:0
5
-
6
- # change to list chars of your dataset or use default vietnamese chars
7
- vocab: 'aAร ร€แบฃแบขรฃรƒรกรแบกแบ ฤƒฤ‚แบฑแบฐแบณแบฒแบตแบดแบฏแบฎแบทแบถรขร‚แบงแบฆแบฉแบจแบซแบชแบฅแบคแบญแบฌbBcCdDฤ‘ฤeEรจรˆแบปแบบแบฝแบผรฉร‰แบนแบธรชรŠแปแป€แปƒแป‚แป…แป„แบฟแบพแป‡แป†fFgGhHiIรฌรŒแป‰แปˆฤฉฤจรญรแป‹แปŠjJkKlLmMnNoOรฒร’แปแปŽรตร•รณร“แปแปŒรดร”แป“แป’แป•แป”แป—แป–แป‘แปแป™แป˜ฦกฦ แปแปœแปŸแปžแปกแป แป›แปšแปฃแปขpPqQrRsStTuUรนร™แปงแปฆลฉลจรบรšแปฅแปคฦฐฦฏแปซแปชแปญแปฌแปฏแปฎแปฉแปจแปฑแปฐvVwWxXyYแปณแปฒแปทแปถแปนแปธรฝรแปตแปดzZ0123456789!"#$%&''()*+,-./:;<=>?@[\]^_`{|}~ '
8
-
9
- seq_modeling: seq2seq
10
- transformer:
11
- encoder_hidden: 256
12
- decoder_hidden: 256
13
- img_channel: 256
14
- decoder_embedded: 256
15
- dropout: 0.1
16
-
17
- optimizer:
18
- max_lr: 0.001
19
- pct_start: 0.1
20
-
21
- trainer:
22
- batch_size: 128
23
- print_every: 100
24
- valid_every: 500
25
- test_every: 500
26
- iters: 10000
27
- # where to save our model for prediction
28
- export: weights/train_model.pth
29
- checkpoint: ./checkpoint/checkpoint_model.pth
30
- log: ./train.log
31
- # null to disable compuate accuracy, or change to number of sample to enable validiation while training
32
- metrics: 49228
33
- test_metrics: 28918
34
- pretrained: false
35
-
36
- dataset:
37
- # path to image
38
- data_root: /mnt/disk3/CGGANv2
39
- # path to annotation
40
- train_annotation: datasets/labels/train.txt
41
- valid_annotation: datasets/labels/valid.txt
42
- test_annotation: datasets/labels/test.txt
43
- # path to lmdb datasets
44
- train_lmdb: datasets/lmdb/train
45
- valid_lmdb: datasets/lmdb/valid
46
- test_lmdb: datasets/lmdb/test
47
-
48
- # resize image to 32 height, larger height will increase accuracy
49
- image_height: 32
50
- image_min_width: 32
51
- image_max_width: 512
52
-
53
- dataloader:
54
- num_workers: 12
55
- pin_memory: true
56
-
57
- aug:
58
- image_aug: false
59
- masked_language_model: false
60
-
61
- predictor:
62
- # disable or enable beamsearch while prediction, use beamsearch will be slower
63
- beamsearch: false
64
-
65
- quiet: false
66
-
67
- # for train
68
- pretrain: https://vocr.vn/data/vietocr/vgg_seq2seq.pth
69
-
70
- # url or local path (for predict)
71
- weights: https://vocr.vn/data/vietocr/vgg_seq2seq.pth
72
-
73
- backbone: vgg19_bn
74
- cnn:
75
- # pooling stride size
76
- ss:
77
- - [2, 2]
78
- - [2, 2]
79
- - [2, 1]
80
- - [2, 1]
81
- - [1, 1]
82
- # pooling kernel size
83
- ks:
84
- - [2, 2]
85
- - [2, 2]
86
- - [2, 1]
87
- - [2, 1]
88
- - [1, 1]
89
- # dim of ouput feature map
90
- hidden: 256