Yinjun Harold JIA

VQ-VAE training example(v2) returned NAN loss

4

Dear Team Deepmind, I am really grateful that you shared a vqvae_example with sonnet2. However, when running it, I currently encounter a problem of NAN vqvae loss from the beginning....

模型权重

2

请问哪里可以下载模型权重呀 没有看到scripts/download_all_data.sh这个文件 谢谢！

Hi, Your work is really interesting! However, when I run your code, I found a code block that runs very slow: torsional-diffusion/diffusion/torus.py , line 68 score_norm_ = score( sample(sigma[None].repeat(10000, 0).flatten()),...