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

@@ -68,7 +68,7 @@ def train(opt, show_number = 2, amp=False):
model = Model(opt)
print('model input parameters', opt.imgH, opt.imgW, opt.num_fiducial, opt.input_channel, opt.output_channel,
opt.hidden_size, opt.num_class, opt.batch_max_length, opt.Transformation, opt.FeatureExtraction,
opt.SequenceModeling, opt.Prediction)
opt.SequenceModeling, opt.Prediction, opt.saved_model)
if opt.saved_model != '':
pretrained_dict = torch.load(opt.saved_model)