Tiansheng Wen

icon indicating an email [email protected]

icon company Xidian University icon location Xia’an China icon location Senior at Xidian University. I will be doing my postgraduate studies under the supervision of Pro. Chen Bo, focusing on machine learning and signal processing

Results 2 issues of Tiansheng Wen

Train on Multi-GPU

3 comment

When I try to train this model on multi-gpu using Accelerate, an error occurs ![image](https://github.com/lucidrains/denoising-diffusion-pytorch/assets/56498265/10ebdc98-14f2-4387-8598-09bb65a5f65c) How to solve it and is there any default Accelerate config that I can refer...

How to generate image using this repository

HI, I am a rookie in this area, and I'm learning the original VQ-VAE. According to the original paper, how can I generate images using PixelCNN after finishing training the...