Question about running train.py

[YuyingShang]

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.

[logan-markewich]

"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