如何在tensorflow.js中克隆模型

时间:2019-06-25 12:03:37

标签: tensorflowjs

在根据一组数据训练一个模型之后,我想获得一个副本,并在不同的数据集上训练每个副本。有没有一种方法可以克隆模型,使得克隆后每个模型都可以单独训练?

我已经尝试过将模型保存在本地存储中,并从那里存储一个副本,但是tensorflow抱怨一些变量名已被使用。

1 个答案:

答案 0 :(得分:0)

万一其他人感兴趣,我最终解决了它,它利用getweights()函数获得了我想要复制的模型的所有权重,使用相同的架构创建了一个新模型,然后将其他模型的用setweights()