python - 如何在不枚举的情况下索引数据加载器中的特定对象

Question

我有以下代码用于在 pytorch 中测试网络（取自pix2pix）：

for i, data in enumerate(dataset):
    model.set_input(data)
    model.test()

我尝试将其更改为仅访问一个元素（任何）而不是迭代多个元素。以下：

data=dataset[0]

导致

TypeError：“CustomDatasetDataLoader”对象不支持索引

CustomDatasetDataLoader作者的自定义类在哪里（如果我理解正确的话）从torch.utils.data.DataLoader突出。

我对python相当陌生，dataset如果可能的话，我如何索引单个特定元素？

标签： pythondatasetpytorch