TensorFlow指定每个epoch训练多少个批次的数据

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

model.fit(train_dataset, epochs=3, steps_per_epoch=100)