我是Tensorflow的新手,在我目前的项目中,我无法在预测/推断之后立即计算损失,而是每2或3次预测,所以我在考虑将每个预测的张量保存在列表中并运行他们之后通过优化器。
我是tensorflow的新手并且不熟悉它,所以如果没有办法做到这一点,欢迎解决这个问题的其他方法。
提前感谢您的帮助!
答案 0 :(得分:0)
为什么你可以计算损失? 如果我理解你的问题吧。您的情况与图中分布式模型非常相似。让每个GPU /服务器计算一个批处理然后收集所有推理和丢失,计算它们的平均值,然后更新变量。