Merge pull request #1 from juliagmt-google/juliagmt-google-patch-1

Update run_release_test.sh
pytorch · Oct 1, 2024 · e309f1b · e309f1b
2 parents 4445aa2 + 1d214e5
commit e309f1b
Showing 1 changed file with 6 additions and 6 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 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