当前位置:首页 > 编程笔记 > 正文
已解决

pytorch定义datase多次重复采样

来自网友在路上 168868提问 提问时间:2023-10-12 20:43:05阅读次数: 68

最佳答案 问答题库688位专家为你答疑解惑

有的时候训练需要对样本重复抽样为一个batch,可以按如下格式定义:

class TrainLoader(Dataset):def __init__(self, fns, repeat=1):super(TrainLoader, self).__init__()self.length = len(fns) # 数据数量self.repeat = repeat # 数据重复次数def __getitem__(self, idx):idx = idx % self.length def __len__(self):return self.length * self.repeat           
查看全文

99%的人还看了

猜你感兴趣

版权申明

本文"pytorch定义datase多次重复采样":http://eshow365.cn/6-19230-0.html 内容来自互联网,请自行判断内容的正确性。如有侵权请联系我们,立即删除!