Spaces:
Runtime error
Runtime error
Delete vgg-seq2seq.yaml
Browse files- 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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|