https://1drv.ms/f/s!Ai2nc20bhVzugYI_SABdx2bYq-tp7w 1
I pickle save the heatmap and grid and write code to reproduce it.
I run it in pycharm with below option ticked
image.png755×672 49.8 KB

it will throw an error:
torch.Size([4, 1, 16, 100, 4096])
THCudaCheck FAIL file=/pytorch/aten/src/THC/generic/THCTensorMath.cu line=238 error=77 : an illegal memory access was encountered

but if i run it without the option ticked, the error msg will not shown. Also not shown in debug mode or python xxx.py
I am not sure if there is no error msg or not shown.