RuntimeError: CUDA error: invalid device function 段错误 (核心已转储)

[zhaiyukun]

File "/STMask/layers/modules/track_to_segment_head.py", line 61, in correlate
out_corr = out_corr.view(b, ph*pw, h, w) / x1.size(1)
RuntimeError: CUDA error: invalid device function
段错误 (核心已转储)

Thank you for sharing the code!

when I run the code, I met the error above.

I would really appreciate your help to fix this problem.

Thank your very much~