allenai / deep_qa

A deep NLP library, based on Keras / tf, focused on question answering (but useful for other NLP too)
Apache License 2.0
404 stars 133 forks source link

Enable batch-parallelization on multiple GPUs #305

Open matt-gardner opened 7 years ago

matt-gardner commented 7 years ago

This should be doable with Keras, as shown here: https://medium.com/@kuza55/transparent-multi-gpu-training-on-tensorflow-with-keras-8b0016fd9012. Matt Peters also got this working, though I'm not sure if he used the approach in that blog post, and he had to fork Keras to get it to work correctly.

matt-gardner commented 7 years ago

326