From 637c62e49f279a9109d1c319a4b6a39c62c90a3d Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Mon, 25 Jul 2022 20:33:52 +0800 Subject: [PATCH] set benchmark_train dynamic epoch --- test_tipc/benchmark_train.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/test_tipc/benchmark_train.sh b/test_tipc/benchmark_train.sh index 19c288ff..ca85b909 100644 --- a/test_tipc/benchmark_train.sh +++ b/test_tipc/benchmark_train.sh @@ -156,6 +156,7 @@ else if [ ${precision} = "fp16" ];then precision="amp" fi + epoch=$(set_dynamic_epoch $device_num $epoch) fp_items_list=($precision) batch_size_list=($batch_size) -- GitLab