Performance issues about tf.function

microsoft / AirSim-Drone-Racing-VAE-Imitation

Code associated with our paper "Learning Visuomotor Policies for Aerial Navigation Using Cross-Modal Representations": https://arxiv.org/abs/1909.06993

MIT License

199 stars 57 forks source link

Performance issues about tf.function #10

Open DLPerf opened 1 year ago

DLPerf commented 1 year ago

Hello! Our static bug checker has found a performance issue in imitation_learning/train_bc.py: Python type argument epoch is passed to tf.function decorated function train 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.

Similar issue in cmvae/train_cmvae.py.

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! @sverrejoh @radical