Loading vectors into a GPU

[Saeed Dehqan]

🚀 Feature

Is there any way for loading vectors based on device with torchtext.vocab.Vectors class?

[Joe Cummings]

This can be accomplished with our prototype Vectors class, which will eventually replace the current implementation.

See here:

text/torchtext/prototype/vectors.py

Line 202 in 79100a6

 class Vectors(nn.Module): 

[Saeed Dehqan]

So, we don't need to change the current implementation to do so, right?

[Joe Cummings]

Nope! You can just do:

vectors = Vectors()
vectors.to(device="cuda")

[Saeed Dehqan]

thanks!