search

lisiyao21 / Bailando

Code for CVPR 2022 paper "Bailando: 3D dance generation via Actor-Critic GPT with Choreographic Memory"
Other
382 stars 59 forks source link

what do you mean by "A kind reminder: the quantization code does not fit multi-gpu training"? #49

Open MengHao666 opened 11 months ago

lisiyao21 commented 11 months ago

Hi, thank you for your interest.

The VQVAE part of this code is suitable for training one GPU. If more GPUs are used, you may kindly refer to https://github.com/openai/jukebox.

Thank you!

MengHao666 commented 11 months ago

Do we need explicitly broadcast tensor value of codebook? Do you have some experiment results to support this?