maszhongming / Effective_Extractive_Summarization

Code for ACL 2019 paper: "Searching for Effective Neural Extractive Summarization: What Works and What's Next"
91 stars 15 forks source link

Runtime CUDA error while evaluating #12

Open Deepanshu-beep opened 2 years ago

Deepanshu-beep commented 2 years ago

While running the testing script I am encountering the following error:

File "main.py", line 345, in test(args, 'test') File "main.py", line 249, in test ext_idx = extractor(raw_art_sents) File "/content/Effective_Extractive_Summarization/decoding.py", line 84, in call indices = self._net.extract([article], k=min(n_art, self._max_ext)) File "/content/Effective_Extractive_Summarization/model/extract.py", line 313, in extract enc_out = self._encode(article_sents, sent_nums) File "/content/Effective_Extractive_Summarization/model/extract.py", line 381, in _encode inputs = self._emb_w(enc_sent) + self.sent_pos_embed(sent_pos) RuntimeError: CUDA error: device-side assert triggered

Kindly look into it