Closed zezeze97 closed 3 years ago
您好，我在使用您的代码进行预测时报错： (torch) zhangzr@AI12:~/MASTER-pytorch-main$ python test.py --checkpoint /home/zhangzr/model_output/master_saved/models/MASTER_Default/example_0712_205619/model_best.pth --img_folder /home/zhangzr/table2latex_data/Training_TSR/train_images --width 400 --height 400 --output_folder /home/zhangzr/model_output/test --gpu 1 --batch_size 4 Loading checkpoint: /home/zhangzr/model_output/master_saved/models/MASTER_Default/example_0712_205619/model_best.pth with saved best metric 0.0000 test data size: 43138 steps: 10785 0it [00:00, ?it/s] Traceback (most recent call last): File "test.py", line 128, in predict(args) File "test.py", line 73, in predict outputs, probs = decode_util.greedy_decode_with_probability(model, images, LabelTransformer.max_length, TypeError: greedy_decode_with_probability() got an unexpected keyword argument 'padding_symbol'
Hi, 已修复。
wenwenyu
commented
3 years ago wenwenyu
commented
3 years ago
您好，我在使用您的代码进行预测时报错： (torch) zhangzr@AI12:~/MASTER-pytorch-main$ python test.py --checkpoint /home/zhangzr/model_output/master_saved/models/MASTER_Default/example_0712_205619/model_best.pth --img_folder /home/zhangzr/table2latex_data/Training_TSR/train_images --width 400 --height 400 --output_folder /home/zhangzr/model_output/test --gpu 1 --batch_size 4 Loading checkpoint: /home/zhangzr/model_output/master_saved/models/MASTER_Default/example_0712_205619/model_best.pth with saved best metric 0.0000 test data size: 43138 steps: 10785 0it [00:00, ?it/s] Traceback (most recent call last): File "test.py", line 128, in
predict(args)
File "test.py", line 73, in predict
outputs, probs = decode_util.greedy_decode_with_probability(model, images, LabelTransformer.max_length,
TypeError: greedy_decode_with_probability() got an unexpected keyword argument 'padding_symbol'