Open YuyingShang opened 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.
"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
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.