mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix find yaml bug for benchmark

pull/1318/head
dongshuilong 2021-10-19 06:38:54 +00:00
parent 5cc5dcc2c2
commit e48181b3d9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
benchmark/run_benchmark.sh
View File

@ -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}"