Skip to content

Commit e8162af

Browse files
committed
fix: 🐛 Keep legacy with dynamo=False
1 parent 666dbb2 commit e8162af

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/convert.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,8 @@ def forward(self, x):
3636
input_names=['input'],
3737
output_names=['output'],
3838
dynamic_axes={'input': {0: 'batch_size'}, 'output': {0: 'batch_size'}},
39-
opset_version=11
39+
opset_version=11,
40+
dynamo=False
4041
)
4142
model = onnx.load(output_model_path)
4243
model_simp, check = simplify(model)

0 commit comments

Comments
 (0)