I am training a Model using DETR. My code :

model = Detr(lr=1e-4, lr_backbone=1e-5, weight_decay=1e-4)
 
batch = next(iter(TRAIN_DATALOADER))
outputs = model(pixel_values=batch['pixel_values'], pixel_mask=batch['pixel_mask'])

During the process I am facing this issue:

TypeError                                 Traceback (most recent call last)
Cell In[22], line 4
1 model = Detr(lr=1e-4, lr_backbone=1e-5, weight_decay=1e-4)
3 batch = next(iter(TRAIN_DATALOADER))
4 outputs = model(pixel_values=batch['pixel_values'], pixel_mask=batch['pixel_mask'])

TypeError: list indices must be integers or slices, not str

I tried rephrasing the code in terms of a for loop but the error still persists.