Can you save model trained by a custom layer?

richliao / textClassifier

Text classifier for Hierarchical Attention Networks for Document Classification

Apache License 2.0

1.07k stars 379 forks source link

Can you save model trained by a custom layer? #2

Closed Sandeep42 closed 7 years ago

Sandeep42 commented 7 years ago

I have an issue with saving the trained model using attention layer. It looks like Keras can not save it, do we need to add anything else to the layer to make it save-able?

richliao commented 7 years ago

Sorry I didn't save the model output. I will do that if I pick up this project again.