Transform using multiple GPUs

openai / generating-reviews-discovering-sentiment

Code for "Learning to Generate Reviews and Discovering Sentiment"

https://arxiv.org/abs/1704.01444

MIT License

1.51k stars 379 forks source link

Transform using multiple GPUs #47

Open liormagen opened 6 years ago

liormagen commented 6 years ago

I'm looking for a way to parallel the transform function. I know that TensorFlow supports that but I couldn't manage to implement it properly in the code.

Help anyone?

spate141 commented 6 years ago

@liormagen You got any update on this? I'm trying out same thing on 8 GPUs but looks like model is only using single GPU for transform() method!

CC: @Newmu