diff --git a/userbenchmark/release-test/run_release_test.sh b/userbenchmark/release-test/run_release_test.sh index 68fd4d452..6272a4524 100644 --- a/userbenchmark/release-test/run_release_test.sh +++ b/userbenchmark/release-test/run_release_test.sh @@ -26,34 +26,34 @@ mkdir -p "${RESULT_DIR}/mnist" pushd "${EXAMPLES_DIR}/mnist" export LOG_FILE=${RESULT_DIR}/mnist/result.log export MEM_FILE=${RESULT_DIR}/mnist/result_mem.log -${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10 +${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 3 # run mnist-hogwild mkdir -p ${RESULT_DIR}/mnist_hogwild pushd "${EXAMPLES_DIR}/mnist_hogwild" export LOG_FILE=${RESULT_DIR}/mnist_hogwild/result.log export MEM_FILE=${RESULT_DIR}/mnist_hogwild/result_mem.log -${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10 +${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 3 # run CPU WLM LSTM mkdir -p ${RESULT_DIR}/wlm_cpu_lstm pushd "${EXAMPLES_DIR}/word_language_model" export LOG_FILE=${RESULT_DIR}/wlm_cpu_lstm/result.log export MEM_FILE=${RESULT_DIR}/wlm_cpu_lstm/result_mem.log -${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10 --model LSTM +${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 3 --model LSTM # run GPU WLM LSTM mkdir -p ${RESULT_DIR}/wlm_gpu_lstm pushd "${EXAMPLES_DIR}/word_language_model" export LOG_FILE=${RESULT_DIR}/wlm_gpu_lstm/result.log export MEM_FILE=${RESULT_DIR}/wlm_gpu_lstm/result_mem.log -${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10 --model LSTM --cuda +${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 3 --model LSTM --cuda # run CPU WLM Transformer mkdir -p ${RESULT_DIR}/wlm_cpu_trans pushd "${EXAMPLES_DIR}/word_language_model" export LOG_FILE=${RESULT_DIR}/wlm_cpu_trans/result.log export MEM_FILE=${RESULT_DIR}/wlm_cpu_trans/result_mem.log -${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10 --model Transformer +${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 3 --model Transformer # run GPU WLM Transformer mkdir -p ${RESULT_DIR}/wlm_gpu_trans pushd "${EXAMPLES_DIR}/word_language_model" export LOG_FILE=${RESULT_DIR}/wlm_gpu_trans/result.log export MEM_FILE=${RESULT_DIR}/wlm_gpu_trans/result_mem.log -${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10 --model Transformer --cuda +${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 3 --model Transformer --cuda