AxisCommunications / onnx-to-keras

Convert onnx models exported from pytorch to tensorflow keras models with focus on performace and highleve compatibility.
MIT License
25 stars 13 forks source link

fix reshape assertion #45

Closed sfluegel05 closed 2 years ago

sfluegel05 commented 2 years ago

Describe your changes

I came across a problem where the assertion for op_reshape failed because x.shape[0] was 64 (my batch size) but shape[0] was -1. Since this is not a mismatch, it should pass the assertion.

Checklist before requesting a review