search

microsoft / CodeBERT

CodeBERT
MIT License
2.25k stars 455 forks source link

Stuck after training #51

Closed PedroEstevesPT closed 3 years ago

PedroEstevesPT commented 3 years ago

Hi,

I was running the example described in section CodeBERT/GraphCodeBERT/translation to convert Java to C#.

However, after the training epoch is concluded, the program does not seem to finalize.

a My terminal is stuck in this state without exiting.

Do you have any idea of what might be going on ?

Can this be caused by a wrong package version (transformers,torch) , does my machine not have enough memory to save the trained model ? (Or if I Ctrl+C the model will be saved without any kind of corruption ?). Even if it is one of the two causes mentioned above I think the program should throw an error instead of being stuck forever after training

At the moment I am using: transformers 4.8.1 torch 1.5.0

Thanks a lot

PedroEstevesPT commented 3 years ago

Nevermind, I just did Ctrl+C to interrupt the program and the models were saved uncorrupted and I was able to perform inference.