lsdefine / attention-is-all-you-need-keras

A Keras+TensorFlow Implementation of the Transformer: Attention Is All You Need
702 stars 188 forks source link

Save model to json #8

Open jingyuanz opened 6 years ago

jingyuanz commented 6 years ago

Have you ever tried save/pickle your trained model? It does not seem to work on my side, reporting an error when I use model.to_json.

lsdefine commented 5 years ago

This function is always failed in my environment, so I usually save/load weights only with using code to construct the model.

hangqiu commented 5 years ago

Could you elaborate on how to save the model after training?