STEP_SIZE_TRAIN=train_generator.n//train_generator.batch_size
STEP_SIZE_VALID=valid_generator.n//valid_generator.batch_size

Try setting step=STEP_SIZE_TRAIN or step=STEP_SIZE_VALID This was not required in the previous versions, may be required now.