Tensorflow JS：推理和训练=真等价

Question

我正在尝试使用Tensorflow教程中介绍的Pix2Pix模型。所以我用Python编写了模型，然后将其转换为javascript。

问题是，在推理上，我需要采取某些层的训练路线，而在tf python中，这很容易，我只是在推理上通过了training = true。

output=model(input, training=true)

但是我如何用转换后的json模型在javascript推断中执行“ training = true”？

const output = model.predict(input); //training=true???

是否有模型参数表或其他函数可以做到这一点？

谢谢您的帮助

Answer 1

[遗憾的是，我没有足够的声誉来发表评论。因此，我将其发布为答案。

我有同样的问题。对我来说，我正在尝试在tensorflow JS中评估Pix2Pix网络。该论文描述了应该从样本中计算出batchnorm统计量，而不是使用存储的移动平均值和训练方差。有关更多详细信息，请参见此github问题tfjs/562

如果有人提出建议，将不胜感激！