如果我们的数据量过大,我们需要将数据转化成Dataset
实例,但是这样我们无法指定需要训练的数据,因为所有数据已经被封装在一个迭代器里面,是多个批次数据
所以针对这个问题,fit()
函数中有个参数可以指定我们每轮训练多少个批次的数据
steps_per_epoch
:这个参数的意思就是我们每个epoch训练多少个batch的数据
model.fit(train_dataset, epochs=3, steps_per_epoch=100)
如果我们的数据量过大,我们需要将数据转化成Dataset
实例,但是这样我们无法指定需要训练的数据,因为所有数据已经被封装在一个迭代器里面,是多个批次数据
所以针对这个问题,fit()
函数中有个参数可以指定我们每轮训练多少个批次的数据
steps_per_epoch
:这个参数的意思就是我们每个epoch训练多少个batch的数据
model.fit(train_dataset, epochs=3, steps_per_epoch=100)