问题
今天模型训练,遇到了个bug
先是在dataloder那报了这样一个错
RuntimeError: Caught RuntimeError in DataLoader worker process 0.
然后后面报
RuntimeError: Trying to resize storage that is not resizable
完整错误代码如下
Traceback (most recent call last):
File "train_temp.py", line 100, in <module>
for data in train_dataloader:
File "/data0/thw/anaconda3/envs/Denoising2/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 628, in __next__
data = self._next_data()
File "/data0/thw/anaconda3/envs/Denoising2/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1333, in _next_data
return self._process_data(data)
File "/data0/thw/anaconda3/envs/Denoising2/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1359,

在进行模型训练时遇到Dataloader的RuntimeError,首先尝试调整num_workers的值,但问题未解决。接着发现是由于输入数据的维度不一致导致,修正数据维度后问题得到解决。此外,了解到num_workers与显卡数量无关,而是影响数据加载进程的数量。
最低0.47元/天 解锁文章
2264





