Closed zhly0 closed 6 years ago
loss = criterion(preds, text, preds_size, label_len)
only preds can be cuda type, the other three shoud be cpu paramters
preds torch.Size([65, 8, 5530]) True text torch.Size([78]) False preds_size torch.Size([8]) False length torch.Size([8]) False
maybe this will helps you
Thanks for reply,I will try!
Is anybody encounter this problem?