Improve release benchmark suites with a lower value of epoch (#2482)

Summary:
#2468

Pull Request resolved: #2482

Reviewed By: xuzhao9

Differential Revision: D64139543

Pulled By: atalman

fbshipit-source-id: 2d030c66d856387b6a2451b26c89fd40e79e0e53

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