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

Optimize batch_dot layer #197

Closed nelson-liu closed 7 years ago

nelson-liu commented 7 years ago

Sorry, in my excitement to merge the batch_dot layer I forgot to check if we could save an allocation via TF/Theano broadcasting. Turns out we can, thanks for the suggestion @matt-gardner

nelson-liu commented 7 years ago

looks like semaphore got stuck --- @matt-gardner could you restart the build when you have a chance?