search

stat-ml / TTDE

12 stars 4 forks source link

Running issues #1

Open shuishen112 opened 1 year ago

shuishen112 commented 1 year ago

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 commented 1 year ago

hi! Have you already solved it?

PgLoLo commented 1 year ago

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 commented 1 year ago

Thanks for your feedback. That makes perfect sense!

PgLoLo commented 1 year ago

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!