0
点赞
收藏
分享

微信扫一扫

ARM 按键控制 LED灯,蜂鸣器,风扇

得一道人 2023-10-12 阅读 33

有的时候训练需要对样本重复抽样为一个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           
举报

相关推荐

0 条评论