Skip to content

upgraded tensorflow and handled torch cuda#619

Open
Sathishsathi wants to merge 2 commits into
NVIDIA:masterfrom
Sathishsathi:master
Open

upgraded tensorflow and handled torch cuda#619
Sathishsathi wants to merge 2 commits into
NVIDIA:masterfrom
Sathishsathi:master

Conversation

@Sathishsathi

@Sathishsathi Sathishsathi commented Dec 13, 2023

Copy link
Copy Markdown

This PR includes following changes:

  • Upgraded tensorflow package
  • Changes made in hparams to adapt the new version of tensorflow
  • hparams changes includes removing of contrib from tensorfow (since it is deprecated in new version) and converted the params as dict
  • To train the tacotron locally where cuda is not available, a condition is added. If cuda is not available, torch will executed without cuda. Please find the code for reference.
  • Tested locally and it is working fine. Since it is handled with if condition, it will not affect the existing flow

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