Element-Research / rnn

Recurrent Neural Network library for Torch7's nn
BSD 3-Clause "New" or "Revised" License
941 stars 313 forks source link

Multi-GPU problem #350

Closed gaosh closed 7 years ago

gaosh commented 8 years ago

Hi, all When I try to implement nn.DataParallelTable on nn.Sequencer for multi-GPU training, it gives me errors. Is this package compatible with nn.DataParallelTable? I tried the solution given by multigpu-nce-rnnlm.lua, but using nn.GPU is not efficient. Only one GPU uses a lot of memory, I guess this is because all the grad parameters is saved to this GPU. Is there any suggestions to for Multi-GPU support?