Fix wrong directories in pretrain.sh.

Author: pengyunie

pretrain.sh

@@ -9,27 +9,23 @@ _DIR=$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )
 function process_data() {
         local ds=$1; shift
 
-        ( cd $_DIR/python/
-          rm -rf $_DIR/output/$ds/data
-          python -m roosterize.main extract_data_from_corpus\
-                 --corpus=$_DIR/../math-comp-corpus\
-                 --output=$_DIR/output/$ds/data\
-                 --groups=$ds
-        )
+        rm -rf $_DIR/output/$ds/data
+        python -m roosterize.main extract_data_from_corpus\
+               --corpus=$_DIR/../math-comp-corpus\
+               --output=$_DIR/output/$ds/data\
+               --groups=$ds
 }
 
 function train_model() {
         local ds=$1; shift
 
-        ( cd $_DIR/python/
-          rm -rf $_DIR/output/$ds/model
-          python -m roosterize.main train_model\
-                 --train=$_DIR/output/$ds/data/$ds-train\
-                 --val=$_DIR/output/$ds/data/$ds-val\
-                 --model-dir=$_DIR/output/$ds/model\
-                 --output=$_DIR/output/$ds/data\
-                 --config-file=$_DIR/configs/Stmt+ChopKnlTree+attn+copy.json
-        )
+        rm -rf $_DIR/output/$ds/model
+        python -m roosterize.main train_model\
+               --train=$_DIR/output/$ds/data/$ds-train\
+               --val=$_DIR/output/$ds/data/$ds-val\
+               --model-dir=$_DIR/output/$ds/model\
+               --output=$_DIR/output/$ds/data\
+               --config-file=$_DIR/configs/Stmt+ChopKnlTree+attn+copy.json
 }
 
 function package_model() {
@@ -43,13 +39,11 @@ function package_model() {
 function eval_model() {
         local ds=$1; shift
 
-        ( cd $_DIR/python/
-          rm -rf $_DIR/output/$ds/results
-          python -m roosterize.main eval_model\
-                 --data=$_DIR/output/$ds/data/$ds-test\
-                 --model-dir=$_DIR/output/$ds/model\
-                 --output=$_DIR/output/$ds/results
-        )
+        rm -rf $_DIR/output/$ds/results
+        python -m roosterize.main eval_model\
+               --data=$_DIR/output/$ds/data/$ds-test\
+               --model-dir=$_DIR/output/$ds/model\
+               --output=$_DIR/output/$ds/results
 }
 
 function retrain_all_models() {