tensorflow LSTM stateless改为stateful
tensorflow LSTM stateless改为stateful
训练模型时用stateless,预测时用stateful保持记忆。预测时新建模型,复制训练模型的参数。
for nb, layer in enumerate(model.layers):
model2.layers[nb].set_weights(layer.get_weights())
...
原创
2020-07-23 11:45:47 ·
224 阅读 ·
0 评论