uguisu

I'm facing the same problem.

我也遇到过同样问题。程序自动对数据集进行分割，分成训练集和测试集。这里的参数`batch_size`和`vaildation_split`直接影响训练集和测试集的大小。如果按照默认设置`batch_size = 32`, `vaildation_split = 0.05`则会出现测试集过小，训练中途失败的问题。 我是通过调整参数来解决问题的。例如将设置修改为`batch_size = 16`, `vaildation_split = 0.1`。