Skip to content

Commit 79f5891

Browse files
committed
train.sh - add #! hash first line
1 parent 624eab6 commit 79f5891

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

train.sh

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
#!/usr/bin/env bash
12
###########################################################
23
# Change the following values to train a new model.
34
# type: the name of the new model, only affects the saved file name.
@@ -14,4 +15,6 @@ model_dir=models/${type}
1415

1516
mkdir -p models/${model_dir}
1617
set -e
17-
python3 -u code2vec.py --data ${data} --test ${test_data} --save ${model_dir}/saved_model
18+
python3 -u code2vec.py --data ${data} --test ${test_data} --save ${model_dir}/saved_model
19+
20+
python3 -u code2vec.py --data data/java14m/java14m --test data/java14m/java14m.val.c2v --save models/java14m/saved_model

0 commit comments

Comments
 (0)