testing dataset ok; trainer.py take effects
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user