我在windows上尝试在自己的数据集上训练nano模型时出现了如下错误:
AttributeError: Caught AttributeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "C:\Users\16353\Anaconda3\envs\pytorch_gpu\lib\site-packages\torch\utils\data\_utils\worker.py", line 287, in _worker_loop
data = fetcher.fetch(index)
File "C:\Users\16353\Anaconda3\envs\pytorch_gpu\lib\site-packages\torch\utils\data\_utils\fetch.py", line 52, in fetch
return self.collate_fn(data)
File "C:\Users\16353\Anaconda3\envs\pytorch_gpu\lib\site-packages\torch\utils\data\_utils\collate.py", line 84, in default_collate
return [default_collate(samples) for samples in transposed]
File "C:\Users\16353\Anaconda3\envs\pytorch_gpu\lib\site-packages\torch\utils\data\_utils\collate.py", line 84, in <listcomp>
return [default_collate(samples) for samples in transposed]
File "C:\Users\16353\Anaconda3\envs\pytorch_gpu\lib\site-packages\torch\utils\data\_utils\collate.py", line 53, in default_collate
numel = sum(x.numel() for x in batch)
File "C:\Users\16353\Anaconda3\envs\pytorch_gpu\lib\site-packages\torch\utils\data\_utils\collate.py", line 53, in <genexpr>
numel = sum(x.numel() for x in batch)
AttributeError: 'int' object has no attribute 'numel'
已是github上最新的代码。请问老师这个问题怎么解决?
你 x 是int整数类型 没有'numel'方法