Update run_release_test.sh

pytorch · Jan 21, 2025 · 2080195 · 2080195
1 parent bfd1ff5
commit 2080195
Showing 1 changed file with 26 additions and 26 deletions.
diff --git a/userbenchmark/release-test/run_release_test.sh b/userbenchmark/release-test/run_release_test.sh
@@ -27,34 +27,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 1
+${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 3
+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 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 3 --model 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 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 3 --model LSTM --cuda
+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 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 3 --model 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 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 3 --model Transformer --cuda
+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 3 --model Transformer --cuda