hello everyone,
i got following error, can anyone help me ?
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/1.research/zh-en/iwslt/with_transformer/models/transformer.py’, line 269, in forward
attn_mask=dec_mask)
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/1.research/zh-en/iwslt/with_transformer/models/transformer.py’, line 119, in forward
query_up = shape_projection(self.w_q(q)) # (Bh, trg_L, dim)
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/linear.py’, line 55, in forward
return F.linear(input, self.weight, self.bias)
File ‘/home/wen/anaconda2/lib/python2.7/site-packages/torch/nn/functional.py’, line 837, in linear
output = input.matmul(weight.t())
File ‘/home/wen/anaconda2/lib/python2.7/site-packages/torch/autograd/variable.py’, line 386, in matmul
return torch.matmul(self, other)
File ‘/home/wen/anaconda2/lib/python2.7/site-packages/torch/functional.py’, line 191, in matmul
output = torch.mm(tensor1, tensor2)
RuntimeError: cublas runtime error : the GPU program failed to execute at