estimator.fit giving error

[shubham-scisar]

estimator.fit is giving the error as :

Train on 128 samples, validate on 128 samples
algo-1-55n7o_1  | Epoch 1/1
algo-1-55n7o_1  | Traceback (most recent call last):
algo-1-55n7o_1  |   File "cifar10_keras_sm.py", line 290, in <module>
algo-1-55n7o_1  |     main(args)
algo-1-55n7o_1  |   File "cifar10_keras_sm.py", line 211, in main
algo-1-55n7o_1  |     validation_steps=(num_examples_per_epoch('validation') // args.batch_size), callbacks=[checkpoint])
algo-1-55n7o_1  |   File "/usr/local/lib/python3.6/dist-packages/keras/engine/training.py", line 1239, in fit
algo-1-55n7o_1  |     validation_freq=validation_freq)
algo-1-55n7o_1  |   File "/usr/local/lib/python3.6/dist-packages/keras/engine/training_arrays.py", line 152, in fit_loop
algo-1-55n7o_1  |     outs = fit_function(fit_inputs)
algo-1-55n7o_1  |   File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/keras/backend.py", line 3476, in __call__
algo-1-55n7o_1  |     run_metadata=self.run_metadata)
algo-1-55n7o_1  |   File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/client/session.py", line 1472, in __call__
algo-1-55n7o_1  |     run_metadata_ptr)
algo-1-55n7o_1  | tensorflow.python.framework.errors_impl.OutOfRangeError: End of sequence
algo-1-55n7o_1  | 	 [[{{node IteratorGetNext}}]]

I am running the code on tensorflow 1.15.2 on tensorflow script mode in sagemaker notebook