可以加载自己的数据集，对原始数据代码部分做适当修改：当你的数据防止在该文件即可。

也可添加微信公众号：AI成长社，回复dmm：可获取当前数据集和。