繁体   English   中英

使用预先训练的举重训练

[英]Training with pre-trained weights

我想使用一组预先训练的权重来为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()开始训练之前,有没有办法可视化起始权重。 在此先感谢!

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

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

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM