Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Changes towards TensorFlow 2.0 compatibility, without TF2 features #154

Open
wants to merge 2 commits into
base: master
Choose a base branch
from

Conversation

gstavrinos
Copy link

Keep in mind that I have not compared the performance of my fixes with the original version (I have access to TF2). Also, in utils/tf_util.py on line 529, where I have replaced the batch_norm from contrib with batch_normalization, some parameters are completely ignored. I am not experienced enough to understand how this could affect performance.

Generally, this is a PR not to for pushing, but to open a discussion on how to proceed.

@gstavrinos
Copy link
Author

Additional comment: in order for the compilation scripts to work with TF2, a symbolic link has to be created, using the following commands:

cd /usr/local/lib/python2.7/dist-packages/tensorflow_core/
sudo ln -s libtensorflow_framework.so.2 libtensorflow_framework.so

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.

None yet

1 participant