quick fix

lsds · May 6, 2019 · 94742e4 · 94742e4
1 parent 959dd83
commit 94742e4
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 9 deletions.
diff --git a/scripts/benchmarks/resnet-50.sh b/scripts/benchmarks/resnet-50.sh
@@ -60,14 +60,12 @@ OPTS="-server -XX:+UseConcMarkSweepGC -XX:NewRatio=2 -XX:SurvivorRatio=16 -Xms48
 CLASS="uk.ac.imperial.lsds.crossbow.ResNetv1"
 CLASSFILE="${TESTS}/`echo ${CLASS} | tr '.' '/'`.class"
 
-#resultdir="results/"
-resultdir="/cache/train_dir"
+resultdir="results/"
 [ ! -d ${resultdir} ] && mkdir -p ${resultdir}
 
 crossbowDirExists ${resultdir}
 
-# datadir="$CROSSBOW_HOME/data/imagenet/"
-datadir="/cache/data_dir/"
+datadir="$CROSSBOW_HOME/data/imagenet/"
 crossbowDirExists ${datadir}
 
 layers=50

diff --git a/scripts/huawei/train.sh b/scripts/huawei/train.sh
@@ -2,13 +2,19 @@
 
 CROSSBOW_HOME=/crossbow
 
-python $CROSSBOW_HOME/scripts/huawei/download_data.py
+cd $CROSSBOW_HOME \
+    && git pull \
+    && mvn package \
+    && cd clib-multigpu \
+    && ./genmakefile.sh \
+    && make -j $(nproc) \
+    && cd ../ \
+    && ./scripts/build.sh
 
-# mv /home/work/user-job-dir/Crossbow-scripts/imagenet-test.metadata $CROSSBOW_HOME/data/imagenet/imagenet-test.metadata
-# mv /home/work/user-job-dir/Crossbow-scripts/imagenet-train.metadata $CROSSBOW_HOME/data/imagenet/imagenet-train.metadata
+python $CROSSBOW_HOME/scripts/huawei/download_data.py
 
-mv /home/work/user-job-dir/Crossbow-scripts/imagenet-test.metadata /cache/data_dir/
-mv /home/work/user-job-dir/Crossbow-scripts/imagenet-train.metadata /cache/data_dir/
+mv /home/work/user-job-dir/Crossbow-scripts/imagenet-test.metadata $CROSSBOW_HOME/data/imagenet/imagenet-test.metadata
+mv /home/work/user-job-dir/Crossbow-scripts/imagenet-train.metadata $CROSSBOW_HOME/data/imagenet/imagenet-train.metadata
 
 bash /home/work/user-job-dir/Crossbow-scripts/resnet-50.sh