【发布时间】:2021-07-28 16:08:57
【问题描述】:
编辑:这与一般的 __getitem__ 方法无关,而是在 Pytorch 数据集子类中使用 __getitem__
我正在尝试实现 Pytorchs 数据集类的使用。
指南例如 here 非常好,但我很难弄清楚 Pytorch 对 __getitem__ 的返回值的要求。在Pytorch documentation 我找不到任何关于它应该返回什么的信息;是否有任何可迭代大小为 2 的对象,例如 [sample,target], (sample,target)?在一些指南中,他们返回一个字典,但他们没有指定它是否必须是一个返回的字典。
【问题讨论】:
-
返回值可以是任何东西(不一定是dict或tuple)。
-
我已经对问题进行了编辑以澄清
-
你可以返回任何你想要的,在大多数情况下你只是返回数据和目标,例如
return images, targets -
@CutePoison
__getitem__在 PyTorch 数据集中是一般的__getitem__- 它没有什么特别之处。 -
这不是一个重复的问题,因为 Pytorch 的数据集
__getitem__有一些特殊性。应该删除重复的标志。