testing dataset ok; trainer.py take effects

This commit is contained in:
HT
2025-07-11 11:08:51 +08:00
parent 185959cf2a
commit cfa52e5f2e
11 changed files with 167 additions and 97 deletions

View File

@@ -0,0 +1,46 @@
number: '0123456789'
experiment_name: '4digit'
symbol: ""
lang_char: ''
train_data: 'all_data'
valid_data: 'all_data/4digit_valid'
manualSeed: 1111
workers: 6
batch_size: 32 #32
num_iter: 3000
valInterval: 5
# saved_model: '' #'saved_models/en_filtered/iter_300000.pth'
svaed_model: 'saved_models/4digit/iter_3000.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