In the source code I linked, the actual net gets sent to cuda in the train.py file, but as I mentioned before, I removed that and replaced it with:
torch.cuda.set_device(0)
net = AutoEncoder(config)
net.to(0)
net = torch.nn.DataParallel(net, device_ids = list_of_gpus)