pau1o-hs/Learned-Motion-Matching

IndexError: tensors used as indices must be long, byte or bool tensors

Closed this issue · 3 comments

Running the python decompressor command with cuda enabled or even not enabled will return this error. Using Python 3.10.9

IndexError: tensors used as indices must be long, byte or bool tensors

image

Hi there! It's working from my side, can you check your CUDA version?

image

@mrtariqkhan @pau1o-hs

I fixed that problem buy type casting to long.
" Xgnd = X[batch.long()].transpose(0, 1)
Ygnd = Ytxy[batch.long()].transpose(0, 1)
Qgnd = Qtxy[batch.long()].transpose(0, 1)
Qgnd_xfm = Qxfm[batch.long()].transpose(0, 1)"

error are gone and script are working fine now.

Awesome, thanks!