DexiNed icon indicating copy to clipboard operation
DexiNed copied to clipboard

Performance issues in the definition of pre_process_binary_cross_entropy, DexiNed-TF2/model.py(P1)

Open DLPerf opened this issue 2 years ago • 1 comments

Hello, I found a performance issue in the definition of pre_process_binary_cross_entropy, DexiNed-TF2/model.py, tf.cast(y, dtype=tf.float32) will be created repeatedly during program execution, resulting in reduced efficiency. I think it should be created before the loop.

The same issues exist in:

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

DLPerf avatar Aug 22 '21 08:08 DLPerf

Hi @DLPerf thanks, I'll fix next week because now I am finishing the last version of DexiNed-Pythorch.

xavysp avatar Sep 09 '21 14:09 xavysp