TensorFlow知道整个计算图,它使用自动微分法找到对于各个变量的损失的梯度值。在运行时会使用梯度下降来更新参数。整个模型的训练可以通过反复地运行内置Optimizer来完成。

optimizer = tf.train.GradientDescentOptimizer(0.01)
train = optimizer.minimize(loss)
for i in range(1000):
  sess.run(train, {x:x_train, y:y_train})

通常训练进程先将训练的模型作为checkpoint文件保存,然后测试进程读取checkpoint文件,恢复模型再验证。

results matching ""

    No results matching ""