From 118d2bf0bc9bbba09d1237d049db2217bf917fef Mon Sep 17 00:00:00 2001 From: FredZCY <39992257+FredZCY@users.noreply.github.com> Date: Thu, 20 Jun 2019 15:48:47 +0800 Subject: [PATCH] Update run_sample.sh --- script/run_sample.sh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/script/run_sample.sh b/script/run_sample.sh index 6d4632b..dcf227c 100644 --- a/script/run_sample.sh +++ b/script/run_sample.sh @@ -10,7 +10,8 @@ else if [ ! $2 ]; then echo "Default is CPU for theano or automatic setting for tensorflow depending on what the backend is using." GPUSET='' - elif [ $2 == 'theano' ]; then + elif [ $2 == 'theano' ] + then GPUSET="THEANO_FLAGS=device=gpu$3,floatX=float32" fi MDL=$1