tensorflow - 使用 loadFrozenModel 后如何修改 tensorflowjs 中的辍学率？

Question

我在tensorflow中训练模型，模型有dropout层。然后我将它转换成tensorflowjs，然后我通过loadFrozenModel()加载它，我可以修改之后的dropout率model=tf.loadFrozenModel吗？

标签： tensorflowtensorflow.js

Answer 1

目前冻结的模型无法进一步训练。您当然可以将它们用作迁移学习任务的基础，但该模型中的变量被冻结并且未标记为可更新。

使用迁移学习，您可以检索 dropout 层之前的层并更改 dropout 层并进一步训练