Closed ashawkey closed 2 years ago
fix issue #7. The warning seems to be caused by the fixed output shape
[1, 19]
(since I don't know the actual model definitions). Changing it to dynamic axis[1, 'seqlen']
will fix the warning.Detailed steps:
import onnx from onnx.tools import update_model_dims m = onnx.load('common.onnx') m = update_model_dims.update_inputs_outputs_dims(m, {'input1': [1, 1, 64, 'image_width']}, {'output': [1, 'seqlen']}) onnx.save(m, 'common.onnx')
thank! I will use this code in the next version
fix issue #7. The warning seems to be caused by the fixed output shape
[1, 19]
(since I don't know the actual model definitions). Changing it to dynamic axis[1, 'seqlen']
will fix the warning.Detailed steps: