search

comtalyst / generative-facial-cosmetics

Using GAN to apply lipstick to the input face automatically and smoothly with TensorFlow. Implemented API with Flask.
https://comtalyst.com/#/pages/GFC
2 stars 1 forks source link

Performance issue in model_engineering/trainers/stylegan.py #14

Open DLPerf opened 1 year ago

DLPerf commented 1 year ago

Hello! Our static bug checker has found a performance issue in model_engineering/trainers/stylegan.py: Python type argument epoch is passed to tf.function decorated function train_step instead of tensors. As there is a for loop, every time the value changes, a new graph will be generated, and that can trigger tf.function retracing warning.

Here is the tensorflow document to support it.

Looking forward to your reply.

DLPerf commented 1 year ago

We are investigating this kind of issues, and your answer will be of great help to our work. Can you take a look? Thank you in advance! @comtalyst