atulkum / pointer_summarizer

pytorch implementation of "Get To The Point: Summarization with Pointer-Generator Networks"
Apache License 2.0
904 stars 243 forks source link

Out of Memory error when trying to load the model #36

Open Shashi456 opened 5 years ago

Shashi456 commented 5 years ago

So when I trained the model initially with coverage off for 500k iterations, now what i would like to do is use this same model and further fine tune it on the model for 500k iterations. What i've seen tho is the moment I mention the model_file_path the model crashes and gives an OOM error even though I have a tonne of memory. The model works just fine if i dont specify the model file path