zoukankan      html  css  js  c++  java
  • 使用DataLoader报错AttributeError: 'int' object has no attribute 'numel'

      在跑这段代码的时候报错

    concat_dataset = ConcatDataset([train_dataset, new_set])
    train_data = DataLoader(concat_dataset, batch_size=batch_size, shuffle=True, num_workers=2, pin_memory=True)
    for data, y in train_data:
      print(data, y)

    Original Traceback (most recent call last):
    File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop
       data = fetcher.fetch(index)
    File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/fetch.py", line 47, in fetch
      return self.collate_fn(data)
    File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/collate.py", line 84, in default_collate
      return [default_collate(samples) for samples in transposed]
    File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/collate.py", line 84, in <listcomp>
      return [default_collate(samples) for samples in transposed]
    File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/collate.py", line 53, in default_collate
      numel = sum([x.numel() for x in batch])
    File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/collate.py", line 53, in <listcomp>
      numel = sum([x.numel() for x in batch])
    AttributeError: 'int' object has no attribute 'numel'

      使用了ConcatDataset合并两个dataset后,进行数据加载的时候报错了,主要原因是合并的两个dataset返回的数据类型不一致,遍历两个dataset可以看到一个返回的是'int',一个返回的是'tensor'。

          

       将两个dataset返回的结果改成一致的,就可以解决问题了。

  • 相关阅读:
    线程的基础知识
    并行编程与PLINQ
    swing组件中数据更新与UI更新的关系
    关于EDT和SwingUtilities的invokeLater和invokeAndWait
    内容页访问母版页页控件的方法
    asp.net 中一般应用处理程序使用session
    正则表达式学习日记
    C#中马赛克算法
    初学C与C在TC2.0上运行的注意事项
    delphi7中两个BT的Bug
  • 原文地址:https://www.cnblogs.com/liwill/p/14944430.html
Copyright © 2011-2022 走看看