Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
speedup caption generator on im2txt (#6255)
* replace the usage of python's built in sort with numpy's argsort, which is faster. Also use the given axes to get the most_likely_words instead of calling enumerate and list (it's slower and more memory conssuming) * correct indentation and redaction on comment for the changes purposes according to cshallue's comment
- Loading branch information