DemisEom / RNNT-pytorch

Implementaion RNN tranceducer
21 stars 7 forks source link

RuntimeError: CUDA error: an illegal memory access was encountered #2

Open cdliang11 opened 4 years ago

cdliang11 commented 4 years ago

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

liuming5467 commented 4 years ago

https://blog.csdn.net/ShuqiaoS/article/details/88978986