为什么需要对dataset 做 dataset.repeat(epochs).batch(batch_size)操作?

来源:5-3 feature_column使用

Tangvivi

2022-02-23

写回答

1回答

正十七

2022-02-28

repeat控制的是数据集可以被遍历多少次,而batch则是将数据组织成一个一个的小组。这些都和模型的训练流程相关。

模型的训练中,正常来说,每次需要载入全部数据计算梯度,但是当数据集比较大的时候,内存放不下,因此,每次参加训练的数据只是一部分,即batch的概念。

模型的训练中,通常会遍历数据集很多遍,每一遍是一个epoch,即repeat的概念。

0
2
Tangvivi
还有一个问题,hidden_units的数量一般是根据什么设置的?
2022-03-02
共2条回复

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程

相似问题