diff --git a/code/README.md b/code/README.md index f8af3af..888ff89 100644 --- a/code/README.md +++ b/code/README.md @@ -1,14 +1,24 @@ -# codebert-training - -This code is used to train CodeBERT +# Magisterka - Patryk Bartkowiak ## Required dependencies for the repository to work properly * [Git](https://git-scm.com/) * [PDM](https://pdm-project.org/en/latest/) ## How to run the training? +### 1. Install PDM +```bash +curl -sSL https://pdm-project.org/install-pdm.py | python3 - +``` +### 2. Go to the code directory and install dependencies ```bash cd ./code pdm install +``` +### 3. Run training code +```bash pdm run_training +``` +or +``` +pdm run src/train_codebert_mlm.py ``` \ No newline at end of file