У меня есть следующий код для тестирования сети в 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, если это вообще возможно?