File ‘/home/wen/1.research/zh-en/iwslt/with_transformer/models/transformer.py’, line 181, in forward
output = self.dropout(self.w_2(self.relu(self.w_1(x))))
File ‘/home/wen/anaconda2/lib/python2.7/site-packages/torch/nn/modules/module.py’, line 325, in call
result = self.forward(*input, **kwargs)
File ‘/home/wen/anaconda2/lib/python2.7/site-packages/torch/nn/modules/conv.py’, line 166, in forward
self.padding, self.dilation, self.groups)
File ‘/home/wen/anaconda2/lib/python2.7/site-packages/torch/nn/functional.py’, line 54, in conv1d
return f(input, weight, bias)
RuntimeError: cublas runtime error : the GPU program failed to execute at /opt/conda/conda-bld/pytorch_1512378360668/work/torch/lib/THC/THCBlas.cu:246
can anyone help me ? thanks a lot.