使用loadFrozenModel之后如何修改tensorflowjs中的辍学率?

时间:2019-03-09 06:13:16

标签: tensorflow tensorflow.js

我在tensorflow中训练模型,模型具有辍学层。然后将其转换为tensorflowjs,然后通过loadFrozenModel()加载它,可以在model=tf.loadFrozenModel之后修改辍学率吗?

1 个答案:

答案 0 :(得分:2)

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

使用转移学习,您可以检索辍学层之前的图层,并更改辍学层并进一步训练