Open zenna opened 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.
edge_158
c_99
FWIW this worked previously on older versions of both Flux and ONNX
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
) ofc_99
has 24 channels but the next layer expects 6 layers.