使用fastai库执行用于图像识别的cnn模型时出现运行时错误

Question

我正在训练一个cnn模型来识别图像。然而，我在运行此代码时遇到了一个错误：

from fastai.vision.all import *
path = untar_data(URLs.PETS)/‘images’
def is_cat(x): return x[0].isupper()
dls = ImageDataLoaders.from_name_func(
path, get_image_files(path), valid_pct=0.2, seed=42,
label_func=is_cat, item_tfms=Resize(224))
learn = cnn_learner(dls, resnet34, metrics=error_rate)
learn.fine_tune(1)

错误：

During handling of the above exception, another exception occurred:
RuntimeError Traceback (most recent call last)
in
----> 1 learn.fine_tune(1)
RuntimeError: DataLoader worker (pid(s) 12456, 4440, 3268, 448) exited unexpectedly

错误发生在最后一行(这是一个较长的错误，但SO不允许我提交所有这些(。我没有在GPU上运行(正如互联网上建议的那样(，因为我还没有真正知道如何告诉jupiter笔记本电脑这样做。

你能帮忙吗？谢谢，路易吉

Answer 1

您可以添加num_workers=0示例

ImageDataLoaders.from_name_func(path, files, label_func, item_tfms=Resize(224),**num_workers=0**)