majianjia / nnom

A higher-level Neural Network library for microcontrollers.
Apache License 2.0
815 stars 235 forks source link

IndexError: list index out of range #181

Open nnilayy opened 1 year ago

nnilayy commented 1 year ago

issue Hello there, So i was trying to use the generate model function for my Unet model, and after sometime i ran into the IndexError: list index out of range. Does anybody know why this error is taking place.

bfs18 commented 1 year ago

change the condition to

        if ('conv' in layer.name) and \
            (len(layer.outbound_nodes) and 'batch_normalization' in layer.outbound_nodes[0].outbound_layer.name):
nnilayy commented 1 year ago

Thank you for your response. I changed the code, but am still facing the same issue.

Issue