加载datalodaer 调用 __getitem__方法验证
最近理了理__getitem__运行原理,该方法返回的是键值为0~n-1的整数,n是__len__方法返回的值;
def getitem(self, index):中的index是在0~n-1不重复随机取值;
import torch.utils.data
class Model():
def __init__(self, animal_list):
self.animal_list = animal_list
# 根据迭代batch_size进行返回
def __getitem__(s
原创
2020-09-05 21:52:00 ·
426 阅读 ·
0 评论