google-coral / edgetpu

Coral issue tracker (and legacy Edge TPU API source)
https://coral.ai
Apache License 2.0
427 stars 125 forks source link

'Internal compiler error. Aborting!', EfficientDet D0 #344

Closed masc-it closed 3 years ago

masc-it commented 3 years ago

Hello, I've successfully trained (with post training quantization, tf 2.4.1) EfficientDet-D0 512x512 and converted it to tflite. As soon as I run !edgetpu_compiler -s "export/model.tflite" I get a way too generic: 'Internal compiler error. Aborting!'.

As I read in other issues, it's not about unsupported operations, maybe something about number of params. It would be great to have a more detailed message though..

In the end, can I do anything about this?

manoj7410 commented 3 years ago

@masc-it Can you please share your model at coral-support@google.com ?

masc-it commented 3 years ago

@manoj7410 sure can! Sending it right now

mbrooksx commented 3 years ago

Hi @masc-it ,

EfficientDet won't directly compile to the EdgeTPU. We've been working with the team for EfficientDet-Lite: https://tfhub.dev/tensorflow/lite-model/efficientdet/lite0/int8/1. This model can be trained and compiled for the EdgeTPU. We are working to add Lite0-Lite4 onto our website as well as instructions for retraining using https://www.tensorflow.org/lite/guide/model_maker.

masc-it commented 3 years ago

@mbrooksx great! So we will be to train it using the classical model.pipeline and tf1 ? Just in case, if you're going to release some beta version of it I'd be available to test it (training or whatever). In general it would be great if you let the open source community test and, in case, help debugging new models (maybe a 'beta' section?)

Naveen-Dodda commented 3 years ago

@masc-it

You can find more details in this page to train EfficientNetDet-lite.

I shall keep you posted if come across any additional details about Det-lite