Pytorch device is not followed in embeddings

When setting a specific device to use for feature extraction in pytorch, the features are still extracted using device "cuda" (i.e., the first GPU).