2020
10-08
10-08
pytorch随机采样操作SubsetRandomSampler()
这篇文章记录一个采样器都随机地从原始的数据集中抽样数据。抽样数据采用permutation。生成任意一个下标重排,从而利用下标来提取dataset中的数据的方法需要的库importtorch使用方法这里以MNIST举例train_dataset=dsets.MNIST(root='./data',#文件存放路径train=True,#提取训练集transform=transforms.ToTensor(),#将图像转化为Tensordownload=True)sample_size=len(train_da...
继续阅读 >