TQDM Progress Bar does not work correctly when initial_epoch is nonzero

[relaxation82]

System information

• OS Platform and Distribution (e.g., Linux Ubuntu 16.04): OS X 10.15.4
• TensorFlow version and how it was installed (source or binary): 2.2.0-rc3, binary
• TensorFlow-Addons version and how it was installed (source or binary): 0.9.1
• Python version: 3.7.7
• Is GPU used? (yes/no): no

Describe the bug

Progress bar of the whole training process over multiple epochs never reaches 100% when the training is restarted, and parameter initial_epoch is therefore nonzero.

Code to reproduce the issue

Essentially

model.fit(x=X, y=y, class_weight=None, batch_size=batchSize, verbose=0, callbacks=tfa.callbacks.TQDMProgressBar(), validation_split=0.2, shuffle=True, epochs=epochCount, initial_epoch=initialEpoch)

where initialEpoch > 0

Other info / logs

The issue should be rather clear with the provided info