From 1d214e57fa2254d294954c299abf661e9b1edb9f Mon Sep 17 00:00:00 2001
From: Julia Guo <153684546+juliagmt-google@users.noreply.github.com>
Date: Tue, 1 Oct 2024 15:18:03 -0700
Subject: [PATCH] Update run_release_test.sh

Update epoch from 10 to 3
---
 userbenchmark/release-test/run_release_test.sh | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/userbenchmark/release-test/run_release_test.sh b/userbenchmark/release-test/run_release_test.sh
index 38e8141d37..8eb7112ac5 100644
--- 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