有人知道从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 []
最终,预测结果与之前相同。
此外,如果可以继续训练已加载的模型,是否可以更新保存的模型的权重和偏置?
回答:
您只是在输入数据并获取损失值;这样是不会训练模型的。为了训练模型,您需要提供数据批次并运行更新操作(可能由optimizer.minimize
返回)。
从Java中进行这些操作是可能的,但Python中的基础设施更加完善,包括用于预取输入数据的线程、监控输入何时结束、保存摘要以及进行分布式训练。