search

FluxML / ONNX.jl

Read ONNX graphs in Julia
Other
139 stars 26 forks source link

Fails on shufflenet #25

Open zenna opened 5 years ago

zenna commented 5 years ago

ONNX's fails on Shufflenet, a network in the ONNX ModelZoo:

https://github.com/onnx/models/tree/master/vision/classification/shufflenet

ONNX.jl converts it to a Flux model fine. However, It fails (on an input of size (224, 224, 3, 1)). It looks like the first error is that the output (edge_158) of c_99 has 24 channels but the next layer expects 6 layers.

zenna commented 5 years ago

FWIW this worked previously on older versions of both Flux and ONNX