Tensorflow 1.12中的自定义培训

时间:2020-10-20 11:52:54

标签: python tensorflow

我正在尝试使用“自定义训练”方法训练模型。我已经成功运行/执行了以下操作:

  1. 制作模型
  2. for循环获取数据。
  3. 初始化Adam优化器。
  4. 获取损失值。

然后问题开始了。在最新版的TF中,我们有apply_gradient方法供优化程序应用从损耗值生成的梯度。但是在TF1.12中不是这样,没有这样的功能。

现在我无法弄清楚应该如何将损失函数中的梯度应用于优化器。

0 个答案:

没有答案