Steps per epoch keras imagedatagenerator. This helped me solve the problem model.