From e48181b3d9007f959a7d70e5dbc1d144c1f9b501 Mon Sep 17 00:00:00 2001
From: dongshuilong <dongshuilong@baidu.com>
Date: Tue, 19 Oct 2021 06:38:54 +0000
Subject: [PATCH] fix find yaml bug for benchmark
---
benchmark/run_benchmark.sh | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/benchmark/run_benchmark.sh b/benchmark/run_benchmark.sh
index 1e51f9a66..85cb4bc81 100644
--- a/benchmark/run_benchmark.sh
+++ b/benchmark/run_benchmark.sh
@@ -21,9 +21,9 @@ function _train(){
echo "current CUDA_VISIBLE_DEVICES=$CUDA_VISIBLE_DEVICES, gpus=$num_gpu_devices, batch_size=$batch_size"
if [ ${fp_item} = "fp32" ];then
- model_config=`find ppcls/configs/ -name ${model_name}.yaml`
+ model_config=`find ppcls/configs/ImageNet -name ${model_name}.yaml`
else
- model_config=`find ppcls/configs/ -name ${model_name}_fp16.yaml`
+ model_config=`find ppcls/configs/ImageNet -name ${model_name}_fp16.yaml`
fi
train_cmd="-c ${model_config} -o DataLoader.Train.sampler.batch_size=${batch_size} -o Global.epochs=${epochs}"