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

Out of Memory #26

Open lucaskingjade opened 2 years ago

lucaskingjade commented 2 years ago

Hi there, When I run the second step as your instructions, I met the "out of memory" problem. I tried to debug the problem and found it is because the music_data is float64 and the memory is consumed rapidly when converting the list music_data to music_np. (in File "utils/functional.py"). Have you ever met the same problem like me? Is it possible to use float32 for training data(music_np) without decreasing the performance of the final model?

BTW: There are 120G of memory in my computer.

lisiyao21 commented 2 years ago

Yes! It is feasible to do so. I am sorry for I did not clean the code sufficiently.