网站首页
编程语言
数据库
系统相关
其他分享
编程问答
pytorch-dataloader
2024-07-25
Pytorch Dataloader 添加批量维度
我认为这个问题已经被问过几次了,但我还没有在这里找到一个好的答案。所以我有一个由2个numpy数组组成的Pytorch数据集。以下是维度。特征=[10000,450,28]numpy数组。dim_0=样本数,dim_1=时间序列,dim_2=特征。基本上我有一个450帧长的数据,其中每
2024-07-24
我想使用 torch DataLoader 并使用生产者/消费者模式,但它卡住了
我想更改torchDataLoader,并在其中使用消费者/生产者模式。我有一个队列,一个线程将文件放入其中,这些项目由框架使用__getitem__使用。这是我的代码:importglobimporttimefromtorch.utils.dataimportDataLoader,Datasetimportthreadingimportqueue