Add export MIOPEN_FIND_MODE=4

train.sh

@@ -4,6 +4,7 @@ echo "Export params ..."
 export HIP_VISIBLE_DEVICES=0,1,2,3 # 自行修改为训练的卡号和数量
 export HSA_FORCE_FINE_GRAIN_PCIE=1
 export USE_MIOPEN_BATCHNORM=1
+export MIOPEN_FIND_MODE=4
 
 echo "Training start ..."
 # coco_path是训练数据集地址，数据是coco format

val.sh

@@ -3,6 +3,7 @@ echo "Testing start ..."
 export HIP_VISIBLE_DEVICES=1
 export HSA_FORCE_FINE_GRAIN_PCIE=1
 export USE_MIOPEN_BATCHNORM=1
+export MIOPEN_FIND_MODE=4
 
 # resume: 待测试模型地址
 # coco_path: 训练数据集地址, 数据是coco format