set benchmark_train dynamic epoch

tink2123 · Jul 25, 2022 · 637c62e · 637c62e
1 parent 00fa71d
commit 637c62e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)