RNNT-pytorch
RNNT-pytorch copied to clipboard
RuntimeError: CUDA error: an illegal memory access was encountered
train_loss = model(inputs, targets_list, input_sizes, target_sizes) File "/home/Liangcd/anaconda3/envs/pyEnv/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in call result = self.forward(*input, **kwargs) File "/home/Liangcd/code/pytorch/RNNT-pytorch-master/models/models.py", line 188, in forward loss = self.loss(out, ys.int(), xlen, ylen) File "/home/Liangcd/anaconda3/envs/pyEnv/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in call result = self.forward(*input, **kwargs) File "/home/Liangcd/anaconda3/envs/pyEnv/lib/python3.7/site-packages/warprnnt_pytorch-0.1-py3.7-linux-x86_64.egg/warprnnt_pytorch/init.py", line 100, in forward return self.loss(acts, labels, act_lens, label_lens, self.blank, self.reduction) File "/home/Liangcd/anaconda3/envs/pyEnv/lib/python3.7/site-packages/warprnnt_pytorch-0.1-py3.7-linux-x86_64.egg/warprnnt_pytorch/init.py", line 40, in forward grads /= minibatch_size RuntimeError: CUDA error: an illegal memory access was encountered
https://blog.csdn.net/ShuqiaoS/article/details/88978986