I solved it like this:
res,*_ = m(torch.tensor([n[0]]).unsqueeze(0).cuda())
Please let me know if this expression can be simplified as I’m completely new to pytorch.
Also, (not yet released) torchtext 0.3 is required for pytorch 0.4
I solved it like this:
res,*_ = m(torch.tensor([n[0]]).unsqueeze(0).cuda())
Please let me know if this expression can be simplified as I’m completely new to pytorch.
Also, (not yet released) torchtext 0.3 is required for pytorch 0.4