Make it compatible with tensorflow-2.5.0

drawbridge / keras-mmoe

A TensorFlow Keras implementation of "Modeling Task Relationships in Multi-task Learning with Multi-gate Mixture-of-Experts" (KDD 2018)

MIT License

690 stars 219 forks source link

Make it compatible with tensorflow-2.5.0 #18

Closed colinyoyo26 closed 3 years ago

colinyoyo26 commented 3 years ago

Originally the requirements.txt couldn't be installed correctly due to version confliction and we can't run the code because of incompatible syntax with tf2.

alvin319 commented 3 years ago

I actually just merged the dependabot's branch to update the TensorFlow version, @colinyoyo26 can you re-submit another PR again based on the master branch? Thanks!

colinyoyo26 commented 3 years ago

of course