-
Jay Zhang authored
* Fix an issue that ShuffleNetV2 model is exported to a wrong ONNX file if dynamic_axes field was provided. * Add a ut for the bug fix. * Fix flake8 issue. * Don't access each element in x.shape, use x.size() instead. Co-authored-by:Vasilis Vryniotis <datumbox@users.noreply.github.com>
45d9a304