清除keras中的优化程序权重

时间:2017-12-10 05:34:02

标签: keras

是否有可能在Keras中清除优化程序权重以重新启动优化程序,因为最近的一些论文表明这实际上产生了良好的结果。

让我们说以下是我的模特。

lstm_model = Sequential()
lstm_model.add(embedding)
lstm_model.add(Bidirectional(LSTM(lstm_size, return_sequences=True, dropout=dropout, recurrent_dropout=rec_drop)))
lstm_model.add(Dense(len(label_dict), activation="softmax", name="prediction_layer"))

optimizer = Adamax(1e-2) 

lstm_model.load_weights(model_dir + "model_20.hdf5")
#reset optimizer weights
lstm_model.fit(....)

1 个答案:

答案 0 :(得分:2)

非常容易与另一个compile

lstm_model.compile(optimizer=Adamax(1e-2), loss = ....)