This is a demo to illustrate how a single NN network using GPUs in series can be trained using nn.GPU decoration and the 'optim' package. This is useful when one GPU does not have enough memory to store and train a single model.
gregjohnso / multigpu-series.torch Goto Github PK
View Code? Open in Web Editor NEWAn example for how to use multiple GPUs for the same model in series.
License: MIT License