• 2024-07-25Pytorch 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