今天模型训练，遇到了个bug
先是报了这样一个错
RuntimeError: Caught RuntimeError in DataLoader worker process 0.
然后后面报
RuntimeError: Trying to resize storage that is not resizable
完整错误代码如下
ERROR - main.py - 2023-08-07 08:42:53,561 - Traceback (most recent call last):
File "main.py", line 404, in main
runner.train()
File "/home/jsun/PycharmProjects/mcvd-pytorch-master/runners/ncsn_runner.py", line 375, in train
for batch, (X, y) in enumerate(dataloader):
File "/opt/anaconda3/envs/vid/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 530, in next
data = self._next_data()
File "/opt/anaconda3/envs/vid/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1224, in _next_data
return self._process_data(data)
File "/opt/anaconda3/envs/vid/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1250, in _process_data
data.reraise()
File "/opt/anaconda3/envs/vid/lib/python3.8/site-packages/torch/_utils.py", line 457, in reraise
raise exception
RuntimeError: Caught RuntimeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/opt/anaconda3/envs/vid/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop
data = fetcher.fetch(index)
File "/opt/anaconda3/envs/vid/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 52, in fetch
return self.collate_fn(data)
File "/opt/anaconda3/envs/vid/lib/python3.8/site-packages/torch/utils/data/_utils/collate.py", line 173, in default_collate
return [default_collate(samples) for samples in transposed] # Backwards compatibility.
File "/opt/anaconda3/envs/vid/lib/python3.8/site-packages/torch/utils/data/_utils/collate.py", line 173, in
return [default_collate(samples) for samples in transposed] # Backwards compatibility.
File "/opt/anaconda3/envs/vid/lib/python3.8/site-packages/torch/utils/data/utils/collate.py", line 138, in default_collate
out = elem.new(storage).resize(len(batch), *list(elem.size()))
RuntimeError: Trying to resize storage that is not resizable
应该怎么办呀？
enumerate(dataloader)
是不是你的数据格式不对，或者数据文件没找到
数据检查一下
【相关推荐】