When `steps_per_execution` is used, the step number shown by a progress bar does not match how many steps were actually executed. See example here: [colab](https://colab.research.google.com/gist/itmo153277/358d490853d9b9cd662dca488834aafe/wrongstepnumber.ipynb) The step number does not match the number of optimizer iterations