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