在使用多个线程加载数据到模型时,报了一个加载错误:

Caught FileNotFoundError in DataLoader worker process 0.

但是并不知道错误在哪里。

因此正确的做法是先排查哪里出现了错误。

既然多线程不报具体错误的位置,那就不使用多线程将num_workers设置为0:

    data_loader = DataLoader(dataset=data_set,
                             batch_size=128,
                             num_workers=0,
                             shuffle=True)

这样再次调试的时候就知道具体是哪行代码报错了,然后修改就可以了。

更多推荐

Caught FileNotFoundError in DataLoader worker process 0.