Open ChiYing212 opened 4 years ago
May I ask which Pytorch version you used?
0.3.1
when I run the main.py, the above problems occurs in line 213 (line 213 is "mask[idx,:seqlen] = torch.Tensor([1]*seqlen)")
Pytorch 0.3.1 shoule be ok, for newer version, just modify seqlen as seqlen.item()
torch 0.3.0b0+591e73e that is my torch version above,l can run correctly.
Hi, I am also trying to use the tool. But I am wondering where the dataset is. I saw you run this model and posted issues. What dataset did you use? Could you please share with me?
mask[idx,:seqlen] = torch.Tensor(torch.tensor([1])*seqlen)"
Check your input data parameter. Make sure it is in 123123 not in [123123]
when I run the main.py, the above problems occurs in line 213 (line 213 is "mask[idx,:seqlen] = torch.Tensor([1]*seqlen)")