Tensorflow 2.0 Upgrade - Githubissues

Optum / retain-keras

Reimplementation of RETAIN Recurrent Neural Network in Keras

Apache License 2.0

83 stars 31 forks source link

Tensorflow 2.0 Upgrade #21

Closed elizabeth-hancock closed 3 years ago

elizabeth-hancock commented 3 years ago

This PR upgrades TF1 imports -> TF2 and updates documentation accordingly. Unfortunately, since the keras_exp package depends on TF1, I've removed this functionality. I will open a subsequent PR for TF2 distributed GPU training. Also, I've added a hot fix by removing the embedding_constraint variable in the Embedding Layer (see issue 15 for more info), but hopefully that can be fixed soon as well.

CLAassistant commented 3 years ago

All committers have signed the CLA.