bug
Something isn't working
help wanted
Extra attention is needed
good first issue
Good for newcomers
#
torchnlp
Here are 2 public repositories matching this topic...
Exploring GloVe Embeddings
-
Updated
Apr 21, 2021 - Python
Improve this page
Add a description, image, and links to the torchnlp topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the torchnlp topic, visit your repo's landing page and select "manage topics."
Expected Behavior
I want to convert torch.nn.Linear modules to weight drop linear modules in my model (possibly big), and I want to train my model with multi-GPUs. However, I have RuntimeError in my sample code. First, I have _weight_drop() which drops some part of weights in torch.nn.Linear (see the code below).
Actual Behavior
RuntimeError: arguments are located on different GPUs at /