Pytorch的DataLoader报错“TypeError: ‘NoneType‘ object cannot be interpreted as an integer“

最新推荐文章于 2025-02-12 17:17:36 发布

樱木之

最新推荐文章于 2025-02-12 17:17:36 发布

阅读量1.2k

点赞数 3

文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/qq_43369406/article/details/136828701

版权

文章描述了一个在使用PyTorchDataLoader时遇到的TypeError，源于在自定义Dataset的__len__方法中返回值为None。解决此问题需确保__len__方法正确返回input-label的长度，如`def__len__(self):returnlen(self.signals)`。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Traceback (most recent call last):
  File "/home/yingmuzhi/SpecML2/dataset_utils.py", line 109, in <module>
    next(iter(dataloader))
  File "/root/anaconda3/envs/env_cp311_ymz/lib/python3.11/site-packages/torch/utils/data/dataloader.py", line 634, in __next__
    data = self._next_data()
           ^^^^^^^^^^^^^^^^^
  File