Using Tensorflow Estimators with Dataset API results in strange steps behavior

Question

I\'m facing some issues regarding the behavior of the training loop of Tensorflow\'s Estimator and Dataset APIs. The code is as follows (tf2.3):