使用预先训练的举重训练

时间:2019-09-10 16:32:33

标签: python tensorflow keras

我想使用一组预先训练的权重来为MNIST分类训练我的模型。更具体地说,我在一个数据集上训练模型。我想将最终权重用作开始权重,以在不同的数据集上训练模型。为此,我使用

{{1}}

我的问题是model.fit()是否将忽略使用model2.set_weights()设置的初始权重。并且如果确实忽略了,那么有一种方法可以确保model2.fit()使用先前获得的权重。另外,在model.fit()开始训练之前,有没有办法可视化起始权重。提前谢谢!

1 个答案:

答案 0 :(得分:2)

执行model2.set_weights时,您更改了model2的权重。就这样。

您可以用相同的方式查看权重:w2 = model2.get_weights()。然后以方便的方式打印w2