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.
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 functiontrain_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.