kedz / nnsum

An extractive neural network text summarization library for the EMNLP 2018 paper "Content Selection in Deep Learning Models of Summarization" (https://arxiv.org/abs/1810.12343).
108 stars 19 forks source link

fixing issues #10

Open alexookah opened 5 years ago

alexookah commented 5 years ago

https://github.com/kedz/nnsum/issues/7 I encountered the same issue as @tlifcen reported. I added an event callback COMPLETED to save the model when training completes in order to run the evaluation. it seems that the ModelCheckpoint does not save model.cuda so embeddings are not available.

https://github.com/kedz/nnsum/issues/4 I also encountered the same problem here, so i just added the fix to this pull request.