From 5375b66cc98ac88aedebb7b30a7cdf1fcdeaf1f4 Mon Sep 17 00:00:00 2001
From: gengdongjie <gengdongjie@huawei.com>
Date: Thu, 30 Apr 2020 11:10:46 +0800
Subject: [PATCH] add relative path support in resnet50_cifar10 example

---
 example/resnet50_cifar10/run_infer.sh | 18 +++++++++++++++---
 example/resnet50_cifar10/train.py     |  3 ++-
 2 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/example/resnet50_cifar10/run_infer.sh b/example/resnet50_cifar10/run_infer.sh
index 5df659275e..14d7faf981 100755
--- a/example/resnet50_cifar10/run_infer.sh
+++ b/example/resnet50_cifar10/run_infer.sh
@@ -20,13 +20,25 @@ then
 exit 1
 fi
 
-if [ ! -d $1 ]
+get_real_path(){
+  if [ "${1:0:1}" == "/" ]; then
+    echo "$1"
+  else
+    echo "$(realpath -m $PWD/$1)"
+  fi
+}
+
+PATH1=$(get_real_path $1)
+PATH2=$(get_real_path $2)
+
+
+if [ ! -d $PATH1 ]
 then 
     echo "error: DATASET_PATH=$1 is not a directory"
 exit 1
 fi 
 
-if [ ! -f $2 ]
+if [ ! -f $PATH2 ]
 then 
     echo "error: CHECKPOINT_PATH=$2 is not a file"
 exit 1
@@ -48,5 +60,5 @@ cp *.sh ./infer
 cd ./infer || exit
 env > env.log
 echo "start infering for device $DEVICE_ID"
-python eval.py --do_eval=True --dataset_path=$1 --checkpoint_path=$2 &> log &
+python eval.py --do_eval=True --dataset_path=$PATH1 --checkpoint_path=$PATH2 &> log &
 cd ..
diff --git a/example/resnet50_cifar10/train.py b/example/resnet50_cifar10/train.py
index c39d1bcf88..4804fc452e 100755
--- a/example/resnet50_cifar10/train.py
+++ b/example/resnet50_cifar10/train.py
@@ -77,7 +77,8 @@ if __name__ == '__main__':
         opt = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), lr, config.momentum,
                        config.weight_decay, config.loss_scale)
 
-        model = Model(net, loss_fn=loss, optimizer=opt, loss_scale_manager=loss_scale, metrics={'acc'})
+        model = Model(net, loss_fn=loss, optimizer=opt, loss_scale_manager=loss_scale, metrics={'acc'}, amp_level="O2",
+                      keep_batchnorm_fp32=False)
 
         time_cb = TimeMonitor(data_size=step_size)
         loss_cb = LossMonitor()