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

Running issues #1

Open
shuishen112 opened this issue Mar 1, 2023 · 4 comments
Open

Running issues #1

shuishen112 opened this issue Mar 1, 2023 · 4 comments

Comments

@shuishen112
Copy link

Great work. When I run your code as suggested in your readme file. I find it is slow to run in a CPU environment. The code is stuck in this line:

initializing bases...
initializing model...
creating permutations...
creating PAsTTSqrOpt...
creating first params...
initializing canonical...
rank1 init for em...
em...

But If I install the GPU version of Jax, it seems that there are some other environmental issues. Could you please share your running environment for GPU. Thanks in advance.

@wrtnew
Copy link

wrtnew commented Mar 17, 2023

hi! Have you already solved it?

@PgLoLo
Copy link
Collaborator

PgLoLo commented Mar 18, 2023

hi! Have you already solved it?

Jax-related libraries have become outdated since the release of TTDE, thus by installing the newest GPU version of Jax everything falls apart. We will soon update all dependencies which will solve the issue. Sorry for the inconvenience and thank you for your interest in our work!

@shuishen112
Copy link
Author

Thanks for your feedback. That makes perfect sense!

@PgLoLo
Copy link
Collaborator

PgLoLo commented Apr 1, 2023

Thanks for your feedback. That makes perfect sense!

We have updated the code and dependencies to resolve the issue. Please feel free to try it out and let us know if you encounter any further problems or have any questions. We appreciate your help in making our project better!

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

No branches or pull requests

3 participants