45 lines
933 B
YAML
45 lines
933 B
YAML
number: '0123456789'
|
|
experiment_name: '4digit'
|
|
symbol: None
|
|
lang_char: None
|
|
train_data: 'all_data'
|
|
valid_data: 'all_data/4digit_valid'
|
|
manualSeed: 1111
|
|
workers: 6
|
|
batch_size: 32 #32
|
|
num_iter: 3000
|
|
valInterval: 10
|
|
saved_model: 'saved_models/4digit/best_accuracy.pth'
|
|
FT: False
|
|
optim: False # default is Adadelta
|
|
lr: 1.
|
|
beta1: 0.9
|
|
rho: 0.95
|
|
eps: 0.00000001
|
|
grad_clip: 5
|
|
#Data processing
|
|
select_data: '4digit_train' # this is dataset folder in train_data
|
|
batch_ratio: '1'
|
|
total_data_usage_ratio: 1.0
|
|
batch_max_length: 34
|
|
imgH: 32
|
|
imgW: 128
|
|
rgb: True
|
|
contrast_adjust: False
|
|
sensitive: True
|
|
PAD: True
|
|
contrast_adjust: 0.0
|
|
data_filtering_off: False
|
|
# Model Architecture
|
|
Transformation: 'TPS'
|
|
FeatureExtraction: 'ResNet'
|
|
SequenceModeling: 'BiLSTM'
|
|
Prediction: 'CTC'
|
|
num_fiducial: 20
|
|
input_channel: 1
|
|
output_channel: 256
|
|
hidden_size: 256
|
|
decode: 'greedy'
|
|
new_prediction: False
|
|
freeze_FeatureFxtraction: False
|
|
freeze_SequenceModeling: False |