Skip to content
#

dataset

Here are 3,890 public repositories matching this topic...

bloodwass
bloodwass commented Jun 17, 2019

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 /

Improve this page

Add a description, image, and links to the dataset topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the dataset topic, visit your repo's landing page and select "manage topics."

Learn more

You can’t perform that action at this time.