Performance issue in the definition of one_hot_encode_image_op, model/utils.py(P1)
DLPerf opened this issue · 2 comments
DLPerf commented
Hello, I found a performance issue in the definition of one_hot_encode_image_op
, model/utils.py, tf.zeros
will be called repeatedly during program execution, resulting in reduced efficiency. I think it should be created before the loop.
Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.
lreiher commented
Hi, feel free to propose a PR. :) Would also be interesting to see whether this yields a performance improvement.
lreiher commented
Closing due to inactivity.