From de93df86f7376e6736de4ec2817e8b518747491b Mon Sep 17 00:00:00 2001
From: wuzewu <wuzewu@baidu.com>
Date: Wed, 16 Jan 2019 10:19:53 +0800
Subject: [PATCH] add image classification example train script

---
 example/image-classification/train.sh | 33 +++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 example/image-classification/train.sh

diff --git a/example/image-classification/train.sh b/example/image-classification/train.sh
new file mode 100644
index 00000000..4495e62c
--- /dev/null
+++ b/example/image-classification/train.sh
@@ -0,0 +1,33 @@
+#!/bin/bash
+set -o nounset
+set -o errexit
+
+script_path=$(cd `dirname $0`; pwd)
+cd $script_path
+
+model_name=ResNet50
+batch_size=32
+data_dir=./dataset
+class_dim=2
+use_gpu=False
+
+while getopts "m:b:c:d:g" options
+do
+	case "$options" in
+        b)
+            batch_size=$OPTARG;;
+        c)
+            class_dim=$OPTARG;;
+        d)
+            data_dir=$OPTARG;;
+		m)
+			model_name=$OPTARG;;
+        g)
+            use_gpu=True;;
+		?)
+			echo "unknown options"
+            exit 1;;
+	esac
+done
+
+python train.py --data_dir=${data_dir} --batch_size=${batch_size} --class_dim=${class_dim} --image_shape=3,224,224 --model_save_dir=output/ --lr_strategy=piecewise_decay --lr=0.1 --model=${model_name} --use_gpu=${use_gpu}
-- 
GitLab