fix shufflenetv1 gpu bug

official/cv/shufflenetv1/gpu_default_config.yaml

@@ -25,7 +25,7 @@ keep_checkpoint_max: 5
 save_ckpt_path: "./"
 save_checkpoint_epochs: 1
 save_checkpoint: True
-amp_level: "O2"
+amp_level: "O0"
 is_distributed: False
 train_dataset_path: ""
 resume: ""

official/cv/shufflenetv1/scripts/run_eval_gpu.sh

@@ -30,11 +30,11 @@ export CUDA_VISIBLE_DEVICES=$DEVICE_ID
 BASEPATH=$(cd "`dirname $0`" || exit; pwd)
 CONFIG_FILE="${BASEPATH}/../gpu_default_config.yaml"
 
-if [ -d "../eval" ]; then
-    rm -rf ../eval
+if [ -d "eval" ]; then
+    rm -rf eval
 fi
-mkdir ../eval
-cd ../eval || exit
+mkdir eval
+cd eval || exit
 
 python ${BASEPATH}/../eval.py \
     --config_path=$CONFIG_FILE \

official/cv/shufflenetv1/train.py

@@ -58,6 +58,8 @@ def train():
         group_size = 1
         context.set_context(device_id=config.device_id)
 
+    if config.device_target == "GPU":
+        context.set_context(enable_graph_kernel=True)
     # define network
     net = ShuffleNetV1(model_size=config.model_size, n_class=config.num_classes)