huawei experiments checkpoint

lsds · May 5, 2019 · c0d3b2a · c0d3b2a
1 parent c807925
commit c0d3b2a
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 4 deletions.
diff --git a/scripts/benchmarks/resnet-50.sh b/scripts/benchmarks/resnet-50.sh
@@ -60,12 +60,14 @@ 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="results/"
+resultdir="/cache/train_dir"
 [ ! -d ${resultdir} ] && mkdir -p ${resultdir}
 
 crossbowDirExists ${resultdir}
 
-datadir="$CROSSBOW_HOME/data/imagenet/"
+# datadir="$CROSSBOW_HOME/data/imagenet/"
+datadir="/cache/data_dir/"
 crossbowDirExists ${datadir}
 
 layers=50
@@ -142,8 +144,7 @@ NCCL_DEBUG=WARN java $OPTS -cp $JCP $CLASS \
     --layers ${layers} \
     --reuse-memory true \
     --direct-scheduling false \
-    --data-directory ${datadir} \
-    &> ${resultdir}/${resultfile}
+    --data-directory ${datadir}
 
 echo "Done"
 

diff --git a/scripts/huawei/train.sh b/scripts/huawei/train.sh
@@ -0,0 +1,15 @@
+#!/bin/bash
+
+CROSSBOW_HOME=/crossbow
+
+python $CROSSBOW_HOME/scripts/huawei/download_data.py
+
+# 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
+
+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/
+
+bash /home/work/user-job-dir/Crossbow-scripts/resnet-50.sh
+
+python $CROSSBOW_HOME/scripts/huawei/upload_data.py