在 Pytorch 中,有没有办法使用类加载特定的torch.utils.data.DataLoader
单个样本?我想用它做一些测试。
教程使用
trainloader = torch.utils.data.DataLoader(...)
images, labels = next(iter(trainloader))
Run Code Online (Sandbox Code Playgroud)
获取一批随机样本。有没有办法使用DataLoader
来获取特定样本?
干杯
shuffle
输入DataLoader
batch_size
计算您要查找的所需样品所属的批次import torch
import numpy as np
import itertools
X= np.arange(100)
batch_size = 2
dataloader = torch.utils.data.DataLoader(X, batch_size=batch_size, shuffle=False)
sample_at = 5
k = int(np.floor(sample_at/batch_size))
my_sample = next(itertools.islice(dataloader, k, None))
print (my_sample)
Run Code Online (Sandbox Code Playgroud)
输出:
tensor([4, 5])
Run Code Online (Sandbox Code Playgroud)
归档时间: |
|
查看次数: |
10774 次 |
最近记录: |