() File "/home/yifu/pytorch/torch/cuda/__init__.py", line 884, in synchronize return torch._C._cuda_synchronize() RuntimeError: CUDA error: an illegal memory access was encountered CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might ...