dkn16 comments

Repositories
Issues
Comments

Results 2 comments of


                                            dkn16

Error when running shared HMC on distributed GPUs with big dataset

> hey [dkn16](https://github.com/dkn16) this issue is still open? I tried to run and it works fine for me. > > ![codeone](https://user-images.githubusercontent.com/58718316/210193358-5fc974c9-d79d-4074-9fa7-77d9dc01e9bf.PNG) Thank you for your response! The code works fine...

Error when running shared HMC on distributed GPUs with big dataset

Here is the snapshot @ADITYADAS1999 . Besides, I created a colab notebook where we can reproduce the error. The link is: https://colab.research.google.com/drive/1H7uhwCAAkenBT5bRut2t9DbPp9XcwnY8?usp=sharing You can see that when the Y.shape==(4096,256), the...