search

jacobkimmel / pytorch_convgru

Convolutional Gated Recurrent Units implemented in PyTorch
MIT License
191 stars 41 forks source link

Multiple GPUs running issue #5

Closed naveenventuri closed 5 years ago

naveenventuri commented 5 years ago

Hi, When i tried to run this code on multiple gpus i got an error saying that tenors are on different GPUs

jacobkimmel commented 5 years ago

Hey @naveenventuri,

You probably forgot to pass a tensor between the GPUs, or didn't move all the GRUCells in the GRU onto the same GPU. Without seeing your code, it's hard to be more helpful.

Thanks! Jacob