Training train-a2d-transformer_v5 error

Garfield-kh / TM2D

[ICCV 2023] TM2D: Bimodality Driven 3D Dance Generation via Music-Text Integration

74 stars 3 forks source link

Training train-a2d-transformer_v5 error #13

Closed mengW6 closed 2 hours ago

mengW6 commented 2 months ago

Hello, a great piece of work. I encountered an error while trying to reproduce the code. During the process of running train-a2d-transformer_v5.py, I encountered a RuntimeError: copy_if failed to synchronize: cudaErrorAssert: device side assert triggered. Do you know why this is happening? If you could help me resolve this, I would greatly appreciate it.

Garfield-kh commented 2 months ago

Hi, thank you for the interest. It seems to be cuda issue, can you try checking the nvidia driver version + cuda version + pytorch version?

mengW6 commented 2 months ago

Thank you :)