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

Tensorflow 2 Migration & Docker Support #15

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

Conversation

lapplislazuli
Copy link

Hi,

here is a PR in accordance with the Apache 2 Licence.

For a reproduction I had to bump some of the versions as my GPUs were too new.
This does not mean your code didn't work as provided, it just doesn't work on 3080Tis.

I summarized the the changes in the README, but in general I ran the migration script from tensorflow and then had to change the cost-function as it moved to tensorflow-addons.
I also added a docker file and a docker compose that runs the standard script.

You might want to cherry pick things, or just close this PR right away :) I just follow the licence.

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