在使用多个线程加载数据到模型时,报了一个加载错误:
Caught FileNotFoundError in DataLoader worker process 0.
但是并不知道错误在哪里。
因此正确的做法是先排查哪里出现了错误。
既然多线程不报具体错误的位置,那就不使用多线程将num_workers设置为0:
data_loader = DataLoader(dataset=data_set,
batch_size=128,
num_workers=0,
shuffle=True)
这样再次调试的时候就知道具体是哪行代码报错了,然后修改就可以了。