From 94742e40487965cc1ab4e7b0d615bba5fd5a5410 Mon Sep 17 00:00:00 2001 From: LUO MAI Date: Mon, 6 May 2019 15:37:13 +0100 Subject: [PATCH] quick fix --- scripts/benchmarks/resnet-50.sh | 6 ++---- scripts/huawei/train.sh | 16 +++++++++++----- 2 files changed, 13 insertions(+), 9 deletions(-) diff --git a/scripts/benchmarks/resnet-50.sh b/scripts/benchmarks/resnet-50.sh index d3ef866..2dba3cc 100755 --- 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 index 2abaf10..e5683e0 100644 --- 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