Skip to content

Commit 0158c63

Browse files
committed
doc: add to the README a way to get the trained (but not release) java14m model
1 parent 16e3071 commit 0158c63

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

README.md

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,21 @@ tar -xvzf java14m_model.tar.gz
7878
##### Note:
7979
This trained model is in a "released" state, which means that we stripped it from its training parameters and can thus be used for inference, but cannot be further trained. If you use this trained model in the next steps, use 'saved_model_iter8.release' instead of 'saved_model_iter8' in every command line example that loads the model such as: '--load models/java14_model/saved_model_iter8'. To read how to release a model, see [Releasing the model](#releasing-the-model).
8080

81+
#### Downloading a trained model () and pursue its training
82+
83+
We provide another trained model but not released (this is why it is much bigger).
84+
85+
```
86+
wget https://zenodo.org/record/2577346/files/trained_model.tar
87+
tar -xvzf trained_model.tar
88+
```
89+
90+
Then use the `--load` CLI interface of `code2vec.py` to pursue the training of this model, _e.g._:
91+
92+
```text
93+
python3 -u code2vec.py --load trained_model/saved_model_iter8 --data ${data} --test ${test_data} --save ${model_dir}/saved_model
94+
```
95+
8196
#### Training a model from scratch
8297
To train a model from scratch:
8398
* Edit the file [train.sh](train.sh) to point it to the right preprocessed data. By default,

0 commit comments

Comments
 (0)