已解决
取Dataset子集(pytorch)
来自网友在路上 187887提问 提问时间:2023-10-30 03:08:38阅读次数: 87
最佳答案 问答题库878位专家为你答疑解惑
取Dataset子集--pytorch
- 1. why
- 2. how
- 3. example
1. why
我们在调试深度学习代码时,常常会遇到数据集太大,导致调试浪费时间的情况,这种情况下,将数据集中的一个子集拿出来用于调试代码,调试成功在用完整的数据集运行代码成为一个可行的方案。
2. how
pytorch中Torch.utils.data.Subset()
函数提供了一个简便的方式,函数如下,indices表示取子集中样本在dataset
中的序号。
indices可以由以下的形式输入:
indices = range(0, 10) # or
indices = [x for x in range(10)]
3. example
trainset = torchvision.datasets.CIFAR10(root='./data', train=True,download=True, transform=transform)
trainset = torch.utils.data.Subset(trainset,[0,1,2,3,4,5,6,7,8,9,10,11])
trainloader = torch.utils.data.DataLoader(trainset, batch_size=4,shuffle=True, num_workers=2)testset = torchvision.datasets.CIFAR10(root='./data', train=False,download=True, transform=transform)
testset = torch.utils.data.Subset(testset,[1,2,3,4])
testloader = torch.utils.data.DataLoader(testset, batch_size=4,shuffle=False, num_workers=2)
查看全文
99%的人还看了
相似问题
- 90.子集II
- 代码随想录算法训练营第二十八天| 78 子集 90 子集|| 93 复原IP地址
- 代码随想录算法训练营第四十一天 | LeetCode 416. 分割等和子集
- 代码随想录 Day35 动态规划04 01背包问题和完全背包问题 LeetCode T416 分割等和子集
- 【C++代码】分割等和子集,目标和,一和零,零钱兑换,动态规划--代码随想录
- 力扣每日一题78:子集
- 取Dataset子集(pytorch)
- 2698 求一个整数的惩罚数 (子集和,DFS)
- 计算机算法分析与设计(18)---回溯法(介绍、子集和问题C++代码)
- 代碼隨想錄算法訓練營|第四十四天|01背包问题 二维、01背包问题 一维、416. 分割等和子集。刷题心得(c++)
猜你感兴趣
版权申明
本文"取Dataset子集(pytorch)":http://eshow365.cn/6-27587-0.html 内容来自互联网,请自行判断内容的正确性。如有侵权请联系我们,立即删除!