Running issues

[shuishen112]

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]

hi! Have you already solved it?

[PgLoLo]

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]

Thanks for your feedback. That makes perfect sense!

[PgLoLo]

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!