如何在不使用枚举的情况下索引dataloader中的特定对象

2024-10-02 16:31:51 发布

您现在位置:Python中文网/ 问答频道 /正文

我有以下代码来测试pytorch中的网络(取自pix2pix):

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

我已经尝试改变它,在迭代多个元素时只访问一个元素(any)。以下内容:

^{pr2}$

导致

TypeError: 'CustomDatasetDataLoader' object does not support indexing

其中CustomDatasetDataLoader是一个custom class by the author,它(如果我理解正确的话)是来自{a3}的疝。在

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


Tags: 代码intest网络元素forinputdata