Performance issue in the definition of train, lessonOne/imgClassifierWeb/execute.py(P1)

[DLPerf]

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 146

Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.