Hello, I found a performance issue in the definition of test_batch, test_batch.py, tf.argmax and tf.reshape will created repeatedly during program execution, resulting in reduced efficiency. I think they should be created before the loop test_batch.
Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.
Hello, I found a performance issue in the definition of
test_batch
, test_batch.py, tf.argmax and tf.reshape will created repeatedly during program execution, resulting in reduced efficiency. I think they should be created before the looptest_batch
.Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.