在训练的时候一直有这个警告,红红的在一堆训信息中间很突兀,之前没去管它,现在决定除掉它!简单查找消息后大概知道是迭代器到了序列末尾没有停止的原因,比如训练集大小是60000,你设置的batchsize是128,那么一个epoch中会有多少个step呢?很容易计算得到60000/128=468.75,这里出现的小数,说明最后一个step的数据不够128,从输出信息可以知道最后跑了是469个step,如下,可以看到警告信息:
那么一下子就想到是不是把batchsize设置成能被训练集大小整除问题就解决了那呢?经过试验,我将batchsize设置成200,跑了300个step是没错,但是warning还是有哇。
再去网上搜寻了一遍,最后试验了这么一种方法:加上预处理数据集的时候repeat()函数设置为空(不要有参数),在fit()函数中加入step_per_epoch参数,设置其等于训练集大小整除于batchsize,如下所示:
再次运行一遍,此时讨厌的红色warning没有啦!
具体原理暂不深究,等以后再遇到相关问题再答。