我正在尝试使用ResNet架构运行pyTorch管道 .
Everithing在Mac笔记本电脑上使用pyTorch官方Docker在CPU上运行 .
在我收到此错误之前,一切似乎都很好:
Batch 23/51
Batch 24/51
Traceback (most recent call last): File "model.py", line 251, in <module>
loss.backward()
File "/opt/conda/lib/python3.6/site-packages/torch/tensor.py", line 93, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph)
File "/opt/conda/lib/python3.6/site-packages/torch/autograd/__init__.py", line 89, in backward
allow_unreachable=True) # allow_unreachable flag
File "/opt/conda/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 178, in handler
_error_if_any_worker_fails()
RuntimeError: DataLoader worker (pid 397) is killed by signal: Killed.
它似乎在第一个时代的第24批之前正常工作 .
数据集使用标准库中的TensorDataset .
任何人都可以给我一个关于发生了什么的暗示吗?
多次尝试给出不同的批号(即使没有随机播放),笔记本电脑资源在整个运行期间看起来很好 .
如果我尝试用 workers=0
作为装载机我得到:
Batch 24/51
Batch 25/51
Batch 26/51
Batch 27/51
Batch 28/51
Killed
非常感谢你 .