"logits must be 2-dimensional" error on TF 1.9 about easy-bert HOT 3 OPEN

First thing is I want to figure out if this is a version compat issue or a general bug:

Is this a problem with TF 1.9 specifically? Has it worked on previous versions but now you need to run it on 1.9 for some other reason?

If it's only on 1.9, did you bump the TF version on both the Python & Java ends?

There's a decent chance this is just fragility in how the input/output nodes in the graph are getting passed between python & java.

from easy-bert.

If you can leave some more specifics about how you're obtaining, saving, and loading the model that'd be helpful in figuring out what's wrong

from easy-bert.

I didn't try for lower versions of TF. It works fine from TF 1.11 onwards but not on TF 1.9. I used your simple code to save model on Python side (I printed TF version). Loaded on Java side and tried to extract embeddings again using your sample code for Java (I printed TF version here as well).
Later I tried with models using your pom entries as well (i.e., without saving-in-python-then-loading-in-java).

I got this exception on both occassions.

from easy-bert.