我已经为数据集类和数据加载器编写了一段代码，我意识到并试图调试KeyError。
问题是，当我试图在本地VScode环境中运行此代码时，而不是像它那样抛出KeyError，它应该无限循环，试图中断单元格不起作用，我必须重新启动整个笔记本。

dataset_train = COCO_Features(
    _df=train_ids,
    _vocab=train_vocab,
    _features=features_map,
)

dataloader_train = torch.utils.data.DataLoader(dataset_train, batch_size=32, shuffle=True,          num_workers=4, collate_fn=caption_collate_fn)

错误应该在此语句上

next(iter(dataloader_train)).shape

我想知道是否有人经历过这种情况，并知道如何解决这个问题。只张贴的片段与错误作为原来的笔记本太大，张贴，我相信无关的实际问题。
我试着查看我安装的所有不同的扩展，但似乎没有什么不寻常的，只是安装了微软的pylance和python扩展。