gather_out_cuda(): Expected dtype int64 for index

Question

csEylLee opened this issue 2 years ago · comments

In PyTorch version, when I run 'network.py', there will be 'gather_out_cuda(): Expected dtype int64 for index'. Is it a bug?

cse_ylLee · Answer 1 · Mon Mar 06 2023 22:26:31 GMT+0800 (China Standard Time)

emmm, I think the difference among different PyTorch version results in it.