Data Parallel for ConvLSTM

[Luoxi Jing]

Hi,Thanks for sharing great work!

I want to train my model with convLSTMcell Layer on multiple GUPs, but how do I do that?
I'd appreciate it if you could give me a tip! Thanks!

[Alfredo Canziani]

I've never used multiple GPUs.
You may want to check PyTorch documentation.