python - 使用预训练的权重进行训练

Question

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

intial_weights = model1.get_weights()
model2 = create_model()
model2.set_weights(initial_weights)
model2.fit(x=x_train59,y=y_train59, epochs=20,callbacks = [cp_callback2])

我的问题是 model.fit() 是否会忽略使用 model2.set_weights() 设置的初始权重。如果它确实忽略了，有没有办法确保 model2.fit() 使用之前获得的权重。此外，有没有办法在 model.fit() 开始训练之前可视化起始权重。提前非常感谢！

标签： pythontensorflowkeras

Answer 1

当你这样做时model2.set_weights，你改变了权重model2。就这样。

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