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 132 forks source link

Easy profiling functionality #309

Open matt-gardner opened 7 years ago

matt-gardner commented 7 years ago

It'd be really nice to know where there are performance bottlenecks in your model. I think tensorflow 1.0 added some stuff that would make this relatively easy to diagnose; can we put in some simple tooling that tells you which layers take up the majority of your computation time?