已解决
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 内容来自互联网,请自行判断内容的正确性。如有侵权请联系我们,立即删除!
- 上一篇: 帆软报表-SQL片段报错处理
- 下一篇: Unity基础课程之物理引擎6-关于物理材质的使用和理解