search

shayneobrien / coreference-resolution

Efficient and clean PyTorch reimplementation of "End-to-end Neural Coreference Resolution" (Lee et al., EMNLP 2017).
https://arxiv.org/pdf/1707.07045.pdf
185 stars 61 forks source link

error when evaluating #9

Open txAnnie opened 5 years ago

txAnnie commented 5 years ago

Evaluating on validation corpus... 217it [12:27, 5.54s/it]Traceback (most recent call last): File "./src/coref.py", line 690, in trainer.train(150) File "./src/coref.py", line 467, in train results = self.evaluate(self.val_corpus) File "./src/coref.py", line 566, in evaluate predicted_docs = [self.predict(doc) for doc in tqdm(val_corpus)] File "./src/coref.py", line 566, in predicted_docs = [self.predict(doc) for doc in tqdm(val_corpus)] File "./src/coref.py", line 595, in predict spans, probs = self.model(doc) File "/home/xtan/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call tracing_state._traced_module_stack.append(self) File "./src/coref.py", line 429, in forward spans, coref_scores = self.score_pairs(spans, g_i, mention_scores) File "/home/xtan/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call tracing_state._traced_module_stack.append(self) File "./src/coref.py", line 347, in forward pairs = torch.cat((i_g, j_g, i_g*j_g, phi), dim=1) RuntimeError: CUDA error: out of memory

I got this error when evaluating on validation corpus.

txAnnie commented 5 years ago

I used GPU to train.

gaoya-J commented 3 years ago

I got this error when evaluating on validation corpus.