Question

有没有人知道在从Tensorflow Python将模型加载到Java后继续训练模型是否可行？我想出了这段代码，但没有用（是的，输出与输入相同）

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中的基础结构更加完善，包括在队列中预取输入数据的线程，监视输入何时结束，保存摘要以及执行分布式培训。

Java - 列车加载张量流模型

1 个答案: