Tf2 training much slower than tf?

RLlib
1

Hi there,

I was comparing tf and tf2 training speed by training a dummy environment. I noticed that the tf2 training is much slower (10x) than the tf training. Is this expected?

Thanks,
Longyu

2

yeah, tf2 is much slower than tf.
do you have eager_tracing turned on like:

–config=’{“framework”: “tf2”, “eager_tracing”: True}’

? that will make things much faster. but still a bit slower than tf I heard.

3

Thanks, this is indeed much faster! Although from documentation it shows eager_tracing only possible if framework=tfe. I used tfe and it is much faster.

1 Like
4

@gjoliver I didn’t know eager mode made such a difference in speed. Do you know why it’s disabled by default?