Running example got topological sort failed with message: The graph couldn't be sorted in topological order. about keras-mmoe HOT 5 CLOSED

Hi! I'm not entirely sure if this is a bug in the code but rather a bug in TF. Have you tried different versions of Keras and Tensorflow?

from keras-mmoe.

I tried keras 2.2.4/2.2.5 and TF 1.13 and 1.14. Both have this issue. Refer to this: tensorflow/tensorflow#24816

@alvin319 When you train the model, do you also have this warning?

from keras-mmoe.

I just tested the code and I am encountering the same warning, but I didn't have any issues during the model inference. Can you share some error logs?

from keras-mmoe.

I froze the session and export the model to Tensorflow *.pb model format. Then load it into Tensorflow. Inference from TF will show the above warning. The inference results are also different.

from keras-mmoe.

I haven't experimented with that yet and unfortunately, this might be a TensorFlow issue. From the issue you've linked, I suspect the culprit might be https://github.com/drawbridge/keras-mmoe/blob/master/mmoe.py#L190. I don't have much time to work on this right now and if you want to dig into this issue and submit a PR, please feel free!

from keras-mmoe.