Showing
1 changed file
with
14 additions
and
4 deletions
| ... | @@ -56,8 +56,15 @@ | ... | @@ -56,8 +56,15 @@ |
| 56 | "#### Download model weights\n", | 56 | "#### Download model weights\n", |
| 57 | "\n", | 57 | "\n", |
| 58 | "Download the two weights of model from the google drive through the gdown module.\n", | 58 | "Download the two weights of model from the google drive through the gdown module.\n", |
| 59 | - "1. [Added model](https://drive.google.com/uc?id=1YrkwfM-0VBCJaa9NYaXUQPODdGPsmQY4) : A model trained Code2NL on Python using pre-trained CodeBERT (Feng at al, 2020).\n", | 59 | + "1. Added model : A model trained Code2NL on Python using pre-trained CodeBERT (Feng at al, 2020).\n", |
| 60 | - "2. [Diff model](https://drive.google.com/uc?id=1--gcVVix92_Fp75A-mWH0pJS0ahlni5m) : A model retrained by initializing with the weight of model (1), adding embedding of the added and deleted parts(`patch_ids_embedding`) of the code." | 60 | + "2. Diff model : A model retrained by initializing with the weight of model (1), adding embedding of the added and deleted parts(`patch_ids_embedding`) of the code.\n", |
| 61 | + "\n", | ||
| 62 | + "Download pre-trained weight\n", | ||
| 63 | + "\n", | ||
| 64 | + "Language | Added | Diff\n", | ||
| 65 | + "--- | --- | ---\n", | ||
| 66 | + "python | 1YrkwfM-0VBCJaa9NYaXUQPODdGPsmQY4 | 1--gcVVix92_Fp75A-mWH0pJS0ahlni5m\n", | ||
| 67 | + "javascript | 1-F68ymKxZ-htCzQ8_Y9iHexs2SJmP5Gc | 1-39rmu-3clwebNURMQGMt-oM4HsAkbsf" | ||
| 61 | ] | 68 | ] |
| 62 | }, | 69 | }, |
| 63 | { | 70 | { |
| ... | @@ -66,9 +73,12 @@ | ... | @@ -66,9 +73,12 @@ |
| 66 | "id": "P9-EBpxt0Dp0" | 73 | "id": "P9-EBpxt0Dp0" |
| 67 | }, | 74 | }, |
| 68 | "source": [ | 75 | "source": [ |
| 76 | + "ADD_MODEL='1YrkwfM-0VBCJaa9NYaXUQPODdGPsmQY4'\n", | ||
| 77 | + "DIFF_MODEL='1--gcVVix92_Fp75A-mWH0pJS0ahlni5m'\n", | ||
| 78 | + "\n", | ||
| 69 | "!pip install gdown \\\n", | 79 | "!pip install gdown \\\n", |
| 70 | - " && gdown \"https://drive.google.com/uc?id=1YrkwfM-0VBCJaa9NYaXUQPODdGPsmQY4\" -O weight/added/pytorch_model.bin \\\n", | 80 | + " && gdown \"https://drive.google.com/uc?id=$ADD_MODEL\" -O weight/added/pytorch_model.bin \\\n", |
| 71 | - " && gdown \"https://drive.google.com/uc?id=1--gcVVix92_Fp75A-mWH0pJS0ahlni5m\" -O weight/diff/pytorch_model.bin" | 81 | + " && gdown \"https://drive.google.com/uc?id=$DIFF_MODEL\" -O weight/diff/pytorch_model.bin" |
| 72 | ], | 82 | ], |
| 73 | "execution_count": null, | 83 | "execution_count": null, |
| 74 | "outputs": [] | 84 | "outputs": [] | ... | ... |
-
Please register or login to post a comment