Estimator模型训练和数据传递

来源：5-11 TF1_自定义estimator

慕姐8158610

2020-05-08

老师好：
1.在1.0中，在定义优化器的后，定义了train_op = optimizer.minimize(loss, global_step = tf.train.get_global_step())，请问在定义了train_op 之后，我们需要训练的参数有哪些呢，为什么需要训练这些参数？然后是我们如何把这些训练参数加入到训练中？在训练过程中如何又知道我们训练了哪些参数？这是三个问题思考了半天没头绪，恳请老师帮忙解答下，感谢！
2.estimator.train(input_fn=…)，这个数据是如何传递的呢。在tf框架中的很多工作都帮我们做了，过程感觉对用户是透明的，学起来感觉有些抽象

写回答

1回答

正十七

2020-05-13

已采纳

train_op需要训练的参数其实可以由minimize函数的参数传进去：https://www.tensorflow.org/api_docs/python/tf/compat/v1/train/AdamOptimizer#minimize，如果没有传，默认就是所有的可以训练的变量。你可以自定义。
input_fn会生成一个dataset，然后estimator会尝试遍历整个dataset，取出数据去进行训练。这些逻辑是对用户透明的。你可以理解成有一个逻辑来遍历数据集，然后将拿到的数据传给网络训练的逻辑。

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

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

1849 学习 · 896 问题

查看课程

相似问题

数据集全量训练，是不是16G的显存不够用了？

回答 1

请问老师tensorflow中keras和estimator这两个API的训练有什么区别？

回答 1

estimator和keras模型比起来优势在哪？

回答 1

estimator

回答 1

关于模型在手机上训练的问题

回答 2

打开慕课网App查看更多内容