Passing the output from the model into generate_detections with params being a dict from EfficientDetLite0Spec config and scales [0.25] (single image input for now), I get
InvalidArgumentError: indices[0,19206] = 19206 is not in [0, 19206) [Op:GatherV2]

in line 146 of automl/efficientdet/keras/postprocess.py (I can’t post git links yet)
Not sure what I’m doing wrong at this point.
For reference snippets of my code
keras_model = model.create_serving_model()

res = keras_model(np.expand_dims(npimages[0], 0))

detections = generate_detections(params, res[0], res[1], [0.25], [1])