MaskRCNN during training throwing segmentation key error

[ArshadIram]

I am using detectron2 first time for the mask- RCNN. I am using an official notebook (https://public.roboflow.ai/object-detection/bccd).

I did not change anything in the provided notebook except the dataset. When I start training it, I get this error message.

KeyError Traceback (most recent call last)
in
84 patience_counter = 0
85 with EventStorage(start_iter) as storage:
---> 86 for data, iteration in zip(data_loader, range(start_iter, max_iter)):
87 storage.iter = iteration
88

10 frames
/usr/local/lib/python3.7/dist-packages/detectron2/data/detection_utils.py in (.0)
398
399 if len(annos) and "segmentation" in annos[0]:
--> 400 segms = [obj["segmentation"] for obj in annos]
401 if mask_format == "polygon":
402 try:

KeyError: 'segmentation'

Any help will be appriciated.