Skip to content

Add compatibility code for tensorflow 2.0 - #8

Open
bobvo23 wants to merge 2 commits into
seth814:masterfrom
bobvo23:master
Open

Add compatibility code for tensorflow 2.0#8
bobvo23 wants to merge 2 commits into
seth814:masterfrom
bobvo23:master

Conversation

@bobvo23

@bobvo23 bobvo23 commented Oct 26, 2020

Copy link
Copy Markdown

Colab and many people have moved to Tensorflow 2.0 environments, so I have added some compatibility code to run on TF2.
I have tested on Colab, tensorflow 2.3

The changes are quite simple, and they follow Google guidance:
Change all:
import tensorflow as tf
To:
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant