Java - 训练加载的张量流模型

Question

有谁知道在将模型从 Tensorflow Python 加载到 Java 后是否可以继续训练模型？ 我想出了这个sn-p的代码，但是没有用（是的，输出和输入一样）

for(int i = 0; i < 10000; i++) {
    Tensor cost = b.session().runner().feed("input", input).feed("output", input).fetch("cost").run().get(0);
    System.out.println(result1);
}

这是打印 10000 次的内容：

FLOAT tensor with shape []

毕竟，预测和以前一样。

此外，如果可以继续训练加载的模型，是否可以更新保存的模型的权重和偏差？

Answer 1

您正在输入输入并获取损失；这不会训练模型。为此，您需要提供批量数据并运行更新操作（可能从optimizer.minimize 返回）。

从 Java 中可以做到这一点，但 python 中的基础设施更完善，包括预取队列中输入数据的线程、监控输入何时结束、保存摘要和进行分布式训练。