Tensorflow 2 Object Detection API: Error loading TFRecord images (coco)

I'm trying to run object detection model from one of the config files

ssd_mobilenet_v2_320x320_coco17_tpu-8.config

Coco dataset is downloaded from tensorflow-dataset api
https://github.com/tensorflow/datasets

which is bunch of tfrecord files.

when running below command,

PIPELINE_CONFIG_PATH=./ssd_mobilenet_v2_320x320_coco17_tpu-8.config
MODEL_DIR=checkpoints
python object_detection/model_main_tf2.py \
    --pipeline_config_path=${PIPELINE_CONFIG_PATH} \
    --model_dir=${MODEL_DIR} \
    --alsologtostderr

I got the error in loading tfrecord files (saying can not decode file)

tensorflow.python.framework.errors_impl.InvalidArgumentError: 4 root error(s) found.
  (0) Invalid argument:  assertion failed: [Unable to decode bytes as JPEG, PNG, GIF, or BMP]
         [[{{node case/cond/else/_10/case/cond/cond_jpeg/else/_105/case/cond/cond_jpeg/decode_image/cond_jpeg/else/_121/case/cond/cond_jpeg/decode_image/cond_jpeg/cond_png/else/_140/case/cond/cond_jpeg/decode_image/cond_jpeg/cond_png/cond_gif/else/_159/case/cond/cond_jpeg/decode_image/cond_jpeg/cond_png/cond_gif/Assert/Assert}}]]
         [[MultiDeviceIteratorGetNextFromShard]]
         [[RemoteCall]]
         [[IteratorGetNext_1]]
         [[Func/replica_2/Loss/classification_loss_1/write_summary/summary_cond/then/_110/input/_413/_7548]]
  (1) Invalid argument:  assertion failed: [Unable to decode bytes as JPEG, PNG, GIF, or BMP]
         [[{{node case/cond/else/_10/case/cond/cond_jpeg/else/_105/case/cond/cond_jpeg/decode_image/cond_jpeg/else/_121/case/cond/cond_jpeg/decode_image/cond_jpeg/cond_png/else/_140/case/cond/cond_jpeg/decode_image/cond_jpeg/cond_png/cond_gif/else/_159/case/cond/cond_jpeg/decode_image/cond_jpeg/cond_png/cond_gif/Assert/Assert}}]]
         [[MultiDeviceIteratorGetNextFromShard]]
         [[RemoteCall]]
         [[IteratorGetNext_1]]
  (2) Cancelled:  Function was cancelled before it was started
  (3) Invalid argument:  assertion failed: [Unable to decode bytes as JPEG, PNG, GIF, or BMP]
         [[{{node case/cond/else/_10/case/cond/cond_jpeg/else/_105/case/cond/cond_jpeg/decode_image/cond_jpeg/else/_121/case/cond/cond_jpeg/decode_image/cond_jpeg/cond_png/else/_140/case/cond/cond_jpeg/decode_image/cond_jpeg/cond_png/cond_gif/else/_159/case/cond/cond_jpeg/decode_image/cond_jpeg/cond_png/cond_gif/Assert/Assert}}]]
         [[MultiDeviceIteratorGetNextFromShard]]
         [[RemoteCall]]
         [[IteratorGetNext_1]]
         [[loss_shape_1_198795/_2]]
0 successful operations.
1 derived errors ignored. [Op:__inference__dist_train_step_369044]