Hello, I found a performance issue in the definition of train, lessonOne/imgClassifierWeb/execute.py, tf.argmax(softmax_propabilities, axis=1) will be created repeatedly during program execution, resulting in reduced efficiency. I think it should be created before the loop.
The same issue exists in:
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
train, lessonOne/imgClassifierWeb/execute.py,tf.argmax(softmax_propabilities, axis=1)will be created repeatedly during program execution, resulting in reduced efficiency. I think it should be created before the loop.The same issue exists in:
tf.argmax(softmax_propabilities, axis=1)in line 146Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.