![](/img/trans.png)
[英]Why does my training loss oscillate while training the final layer of AlexNet with pre-trained weights?
[英]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.