search

nanguoshun / LSR

Pytorch Implementation of our ACL 2020 Paper "Reasoning with Latent Structure Refinement for Document-Level Relation Extraction"
127 stars 22 forks source link

Question about running train.py #39

Open YuyingShang opened 3 years ago

YuyingShang commented 3 years ago

when I run the code, no matter lsr-bert model or lsr model, after the evaluate_epoch=30, my process will be killed. For both models, I have set my batch_size=3, lr=1e-3. Do you have any idea about this problem? Looking forward to your reply.

image

image

logan-markewich commented 3 years ago

"exit code 137" means out of memory, so you don't have enough memory to train.

I would decrease the batch size to 1 or 2