Caught KeyError in DataLoader worker process 0.

1.问题描述

做目标检测网络训练时，遇到下面报错：
Caught KeyError in DataLoader worker process 0.
KeyError：Sa**（自己数据集的一种类别）

2.解决办法

最后发现是因为类别字母大小写造成的，全部改成小写即可。