Pytorch Dataloader内部随机数

pytorch的dataloader的__getitem__()方法,内部定义随机数,采用numpy.random.rand的方法,每次迭代从dataloader里取数据,都会产生拿到相同的随机数序列,相反,torch.rand的方法,每次迭代拿数据时,拿到的是不同的随机数序列。

	# numpy生成随机数
    def get_train_sample(self, ):
        track_idx = np.random.randint(0, len(self._tracks), (1, ))[0]
        separated = self._tracks[track_idx]
        patch_length = int(self._sample_length * self._sample_rate)
        if self._incoherent_rate <= np.random.rand(1)[0]:   # coherent remix
            start_t = np.random.randint(0, separated.shape[-1]-patch_length, (1, ))
            separated = separated[:, :, start_t: start_t+patch_length]
		return separated
	# torch生成随机数
    def get_train_sample(self, ):
        track_idx = torch.randint(0, len(self._tracks), (1,)).item()
        separated = self._tracks[track_idx]
        patch_length = int(self._sample_length * self._sample_rate)
        if self._incoherent_rate <= np.random.rand(1)[0]:   # coherent remix
            start_t = torch.randint(0, separated.shape[-1]-patch_length, (1, )).item()
            separated = separated[:, :, start_t: start_t+patch_length]
		return separated

用第一种方法生成的数据片段,每个loader里面都相同,用第二种方法,每个loader依旧是随机。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值